torch.cuda.is_available() 返回 false Ubuntu 16.04
torch.cuda.is_available() returning false with Ubuntu 16.04
我有以下配置
- 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？
我在 Ubuntu 18.04 安装 anaconda 时遇到了同样的问题，从 PyTorch 官网推荐。安装 cudatoolkit 10.1 而不是推荐的 10.2 后，我得到了 'True'。有些人使用驱动程序版本 downgrade/upgrade 解决了这个问题，因为这很可能是兼容性问题。
我有以下配置
- 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？
我在 Ubuntu 18.04 安装 anaconda 时遇到了同样的问题，从 PyTorch 官网推荐。安装 cudatoolkit 10.1 而不是推荐的 10.2 后，我得到了 'True'。有些人使用驱动程序版本 downgrade/upgrade 解决了这个问题，因为这很可能是兼容性问题。