报错原因
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