Installation 使用预安装的CUDA和cudnn安装Pytorch GPU

正如标题所示，我已经预装了CUDA和cudnn（我的Tensorflow正在使用它们）

CUDA的版本是10.0来自

nvcc--version

cudnn的版本为7.4

我正在尝试使用

conda install pytorch torchvision cudatoolkit=10.0-c pytorch

在conda环境中安装pytorch

但是，安装的pytorch无法成功检测到我的GPU

有人知道有没有办法用特定的CUDA和cudnn版本安装GPU版本的pytorch？我不想更改CUDA和cudnn版本，因为我的Tensorflow正在使用它们

---

任何想法都将不胜感激

所以我最终自己解决了这个问题。问题是我在安装pytorch后没有重新启动系统。重新启动后，

torch.cuda.is_available（）

按预期返回

True

如果您在终端中键入

nvidia smi

，GPU是否可识别？@JosephKonanYea，我对此没有异议。我还可以在我的系统中使用Tensorflow检测我的GPU。与您正在使用的cuda版本相比，显示器驱动程序版本可能太旧。显示

nvidia smi

输出会有所帮助。要获得更详细的评估，您可以安装numba并执行刚刚使用

nvidia smi

检查的@JosephKonanI，这表明我使用的是430.50。这应该不是问题。