torch.cuda.is_available（）在Ubuntu 16.04中返回false

Question

我有以下配置

• nvcc -V

nvcc: NVIDIA (R) Cuda compiler driver
Copyright (c) 2005-2015 NVIDIA Corporation
Built on Tue_Aug_11_14:27:32_CDT_2015
Cuda compilation tools, release 7.5, V7.5.17

• nvidia-smi

• CUDA工具包（来自Anaconda）

cudatoolkit 10.2.89 hfd86e86_1 anaconda

• PyTorch（来自Anaconda）

pytorch 1.5.0 py3.7_cuda10.2.89_cudnn7.6.5_0 pytorch

但是我得到torch.cuda.is_available() -> False

任何人都可以告诉我升级或降级哪个组件以显示CUDA吗？

Answer 1

在PyTorch官方网站推荐的安装有anaconda的Ubuntu 18.04上，我遇到了同样的问题。在安装cudatoolkit 10.1而不是推荐的10.2之后，我得到了'True'。有些人使用驱动程序版本降级/升级来解决此问题，因为这很可能是兼容性问题。