我正在PyTorch上运行CNN。 torch.cuda.is_available（）函数返回false，并且未检测到GPU。但是，我可以使用GPU运行Keras模型。这是我的系统信息：
当我检查nvidia-smi时，输出显示CUDA版本为10.1。但是，nvcc -V命令告诉我它是CUDA 9.1。
我从官方网站下载了NVIDIA-Linux-x86_64-430.50.run，并使用命令行进行安装。我使用官方网站推荐的以下命令行安装了CUDA 10.1：
replaceBetween('Hi World', 3, 7, 'People');
// Hi People
我通过pip安装安装了PyTorch。怎么了？预先感谢！
答案 0 :(得分：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说明here.（请注意，在安装新的CUDA之前，您可能必须先删除其先前的安装。）