python - 安装 Pytorch cuda 后，torch.cuda.is_available() 显示 false。该怎么办？

Question

我已经通过运行以下命令安装了 pytorch cuda:

conda install pytorch torchvision torchaudio cudatoolkit=11.1 -c pytorch -c nvidia

我的cuda版本是11.2。我正在使用 windows 10 。 Pytorch cuda 11.2 目前不可用。 (pytorch.org)

所以我已经安装了 11.1 版本。

(使用nvidia-smi) Cuda version

但它显示错误。

torch.cuda.is_available() >>> False

我尝试过 10.2 和 11.1 版本。

据我所知，我不需要为pytorch安装cuda工具包

Answer 1

您不应该将软件包安装到您的基础环境中。使用必要的工具创建一个单独的环境。

示例:使用 Python v3.7 和 torch 包创建名为 dlearn 的环境

conda create -n dlearn python=3.7 pytorch torchvision torchaudio cudatoolkit=11.1 -c pytorch -c nvidia

激活并使用您的 dlearn 环境

conda activate dlearn
python -c "import torch;print(torch.cuda.is_available())"
# this should echo True if all is well

目前支持的 cudatoolkit 是 11.1，可以与 11.2 驱动程序配合使用。他们迟早会更新的。如果需要，您可以从 GitHub PyTorch 进行构建以获取最新版本。步骤比上面更复杂。