Pytorch Pytork在检测CUDA时遇到问题

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

• 操作系统：Ubuntu 18.04.3
• Python 3.7.3（Conda）
• GPU:GTX1080Ti
• Nvidia驱动程序：430.50

当我检查nvidia smi时，输出显示CUDA版本为10.1。然而，nvcc-V命令告诉我它是CUDA9.1

我下载了NVIDIA-Linux-x86_64-430.50。从官方网站运行并使用命令行安装。我使用官方网站推荐的以下命令行安装了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安装安装了PyTorch。怎么了？提前谢谢

默认的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，并使用

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）。

我刚刚安装了CUDA 10.0，nvcc仍然告诉我我的版本是9.1。这是我完成安装时的信息。这与我安装CUDA 10.1时的时间相同：===============摘要==========驱动程序：未选择的工具包：安装在/usr/local/CUDA-10.0中示例：安装在/home/kaiming中请确保-PATH包括/usr/local/CUDA-10.0/bin-LD_-LIBRARY\u路径包括/usr/local/CUDA-10.0/lib64，或者，将/usr/local/cuda-10.0/lib64添加到/etc/ld.so.conf并以root用户身份运行ldconfig请询问与安装cuda相关的新问题。您可能需要删除以前的安装。我只需要在CUDA安装后重新安装PyTorch。我真傻。