torch.version.cuda 在某些版本的 pytorch 和 cuda 中为 None

Question

我的cuda版本显示在这里。 nvcc -V 输出 nvidia-smi 输出

我基本上想安装apex。我首先使用命令

conda install pytorch=1.10.1 cudatoolkit=11.1 -c pytorch

用cuda安装torch，这个版本的cudatoolkit工作正常并且

torch.version.cuda 

正确显示预期输出“11.1”并且

torch.cuda.is_available()

返回 True。然而，当我稍后尝试安装 apex 时，这不起作用，因为我的 nvcc -V 的版本如上所示是 11.4，但 cudatoolkit 是 11.1。

然后我使用命令

conda install pytorch cudatoolkit=11.4 -c pytorch

安装正确的 cudatoolkit 版本 11.4。这样做之后，torch版本变为1.9.1。然而，

torch.version.cuda 

变为无并且

torch.cuda.is_available()

返回 False。

我还尝试了 cudatoolkit=11.4 和 torch 版本 1.9.1 和 1.12.1，但它们都不起作用。

我希望我的 GPU 能够被正确检测到。我应该使用哪个版本的 torch 和 cudatoolkit？或者这里还有什么问题吗？多谢。

Answer 1

我遇到了同样的问题，使用 pip 安装解决了我的问题。

    pip3 install torch==1.10.1+cu113 torchvision==0.11.2+cu113 torchaudio==0.10.1+cu113 -f https://download.pytorch.org/whl/cu113/torch_stable.html