相关疑难解决方法(0)

我在 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。怎么了？提前致谢！