报错原因

cuda版本和 pytorch不匹配

解决办法

点击此连接 进入如下图所示的页面

查看cuda版本:nvidia-smi

根据自己所需的torch、系统、cuda版本选择对应的安装方式即可

查看各版本对应关系

1.这里pytorch和cudatoolkit版本对应关系:
https://pytorch.org/get-started/previous-versions/

2.cudatoolkit版本和系统cuda对应关系:
https://www.zhihu.com/question/344950161/answer/818139888

3.系统cuda和nvidia对应关系:
https://blog.csdn.net/He_9520/article/details/100032803

4.cuda和cuDNN的关系和对应关系:
https://www.jianshu.com/p/622f47f94784
https://www.cnblogs.com/yeran/p/11345990.html

查看显卡版本:
ubuntu-drivers devices
nvidia-smi

查看服务器的cuda版本
使用如下命令:
nvcc -V

参考链接

posted on 2021-06-17 11:35  jaysonteng  阅读(22227)  评论(1)    收藏  举报