python - Pytorch 1.4 是否支持 cuda10.1？

Question

我尝试安装pytorch 1.5+cu101，但是找不到caffe2_nvrtc.dll（确实存在于lib\caffe2_nvrtc.dll）。所以我尝试安装 pytorch 1.4+cu101，但我从https://download.pytorch.org/whl/torch_stable.html找不到这样的轮子（只有 pytorch1.4+cu100/cu92）。
有什么帮助吗？

标签： pythonpytorch

Answer 1

PyTorch 1.4.0 默认与 CUDA 10.1 一起提供，因此没有单独的cu101后缀包，这些仅用于替代版本。您只需要安装常规torch软件包：

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