PyTorch 无法检测 CUDA

Question

我在 PyTorch 上运行 CNN。torch.cuda.is_available() 函数返回 false 并且未检测到 GPU。不过，我可以使用 GPU 运行 Keras 模型。这是我的系统信息：

• 操作系统：Ubuntu 18.04.3
• Python 3.7.3（康达）
• 显卡：GTX1080Ti
• Nvidia 驱动程序：430.50

当我检查nvidia-smi时，输出显示CUDA版本是10.1。但是，nvcc -V 命令告诉我它是 CUDA 9.1。

我从官方网站下载了 NVIDIA-Linux-x86_64-430.50.run 并使用命令行安装。我使用官方网站推荐的以下命令行安装了 CUDA 10.1：

wget http://developer.download.nvidia.com/compute/cuda/10.1/Prod/local_installers/cuda_10.1.243_418.87.00_linux.run
sudo sh cuda_10.1.243_418.87.00_linux.run

我通过 pip install 安装了 PyTorch。怎么了？提前致谢！

Answer 1

默认的 Pytorch 1.2 包依赖于 CUDA 10.0，但您有 CUDA 9.1。的输出nvidia-smi只是告诉您 GPU 支持的最大 CUDA 版本，nvcc并给出系统上安装的 CUDA。看来您的 CUDA 10.1 安装不成功。

除了 CUDA 10.0 之外，Pytorch 还支持 CUDA 9.2，并且我发现为 CUDA 10.0 编译的 Pytorch 包也适用于 CUDA 10.1。因此，您可以将 CUDA 安装升级到 9.2 并使用以下命令安装 Pytorch CUDA 9.2 软件包

pip3 install torch==1.2.0+cu92 torchvision==0.4.0+cu92 -f https://download.pytorch.org/whl/torch_stable.html

或者获取 CUDA 10.1 的有效安装。这里有详细的 Linux 说明。（请注意，在安装新的 CUDA 之前，您可能必须删除以前安装的 CUDA。）