pytorch 4
-
Pytorch运行导致炸机重启解决方案
他妈的浪费我两个小时找原因 按照网上的教程一步一步搭yolov3的网络,在搭好测试的那一步(dummy input,都没有训练也没有预加载任何东西),简单的调用model跑一下,输入命令之后两秒内直接黑屏然后重启。 软件环境: win10 专业版 64位 python
-
pytorch与相关环境
第一次搞的时候没有关心过环境这些东西,但是既然重装电脑了,那就考虑认真地配一下环境。CUDA必须要指出的是我不知道怎么在英伟达的页面找到如何下载确定版本的CUDA,https://developer.nvidia.com/cuda-toolkit 上来只有最新版,用Google才能专门找到 CUDA
-
pytorch tricks
tensor的存储方式是行主序,这决定了reshape()的工作方式与matlab不同tensor是有确定维数的,因此tensor(1),tensor([1]),tensor([[1]])是不同的,不能像matlab一样用很高维的索引访问。