如何解决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 举报,一经查实,本站将立刻删除。