我试图安装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.
有什么帮助吗?
PyTorch 1.4。默认情况下,CUDA 10.1附带了0,因此没有带有cu101
后缀的单独软件包,这些软件包仅适用于替代版本。您只需安装常规的torch
软件包:
pip install torch==1.4.0 -f https://download.pytorch.org/whl/torch_stable.html