提问者:小点点

Pytorch 1.4支持cuda10.1吗?


我试图安装pytorch 1.5 cu101,但它找不到caffe2\u nvrtc。dll(它确实存在于lib\caffe2\u nvrtc.dll)。因此,我尝试安装pytorch 1.4 cu101,但我无法从中找到这样的轮子(仅pytorch 1.4 cu100/cu92)https://download.pytorch.org/whl/torch_stable.html.
有什么帮助吗?


共1个答案

匿名用户

PyTorch 1.4。默认情况下,CUDA 10.1附带了0,因此没有带有cu101后缀的单独软件包,这些软件包仅适用于替代版本。您只需安装常规的torch软件包:

pip install torch==1.4.0 -f https://download.pytorch.org/whl/torch_stable.html