微信公众号搜"智元新知"关注
微信扫一扫可直接关注哦!

PyTorch的CUDA版本为10.2,torch_sparse的CUDA版本为11.1,在Ubuntu 20.04上

如何解决PyTorch的CUDA版本为10.2,torch_sparse的CUDA版本为11.1,在Ubuntu 20.04上

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

有人知道我该如何解决这个问题?

完整错误

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

谢谢!

解决方法

问题是您在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。

版权声明:本文内容由互联网用户自发贡献,该文观点与技术仅代表作者本人。本站仅提供信息存储空间服务,不拥有所有权,不承担相关法律责任。如发现本站有涉嫌侵权/违法违规的内容, 请发送邮件至 dio@foxmail.com 举报,一经查实,本站将立刻删除。