PyTorch在Ubuntu 20.04上有CUDA 10.2版本，torch_s稀疏有CUDA 11.1版本

Question

我正在使用Ubuntu 20.04，看起来CUDA10.2不支持最新版本的Ubuntu。

有人知道我该怎么解决这个问题吗？

完整错误：

RuntimeError：检测到PyTorch和torch_s稀疏已编译具有不同的CUDA版本。PyTorch具有CUDA 10.2版本和torch_s稀疏具有CUDA版本11.1。请重新安装torch_s稀疏与您的PyTorch安装相匹配。

谢谢！

Answer 1

问题是Ubuntu中有CUDA 11.1，但Pytorch还不支持这个版本(目前支持的最新版本是10.2(

一个解决方案是降低CUDA:

$ sudo apt remove nvidia-driver-450 
$ sudo sudo apt autoremove 
$ sudo reboot 
$ sudo apt install nvidia-driver-440-server
$ sudo reboot 
$ nvidia-smi 
Thu Oct 15 12:52:24 2020       
+-----------------------------------------------------------------------------+
| NVIDIA-SMI 440.95.01    Driver Version: 440.95.01    CUDA Version: 10.2     |
|-------------------------------+----------------------+----------------------+

计划使用Pytorch 1.7支持CUDA 11。