训练的时候需要的问题,记录一下
如上图所示:这个问题大概率就是GPU版本和pytorch不匹配的问题
1.查看GPU版本
$ nvidia-smi
这里cuda版本是11.7
2. 下载正确的版本号
3. 安装对应的pytorch包,需要到pytorch官网下载。
官网链接: pytorch官网
-c可以去掉,意识是从官网下载,一般下载速度较慢。
tips: pip和conda的区别之处
conda会安装包的其他相关依赖,但是pip并不会检查相关依赖,只会通过错误信息来提醒你!
安装之后重新运行程序就可以解决上诉问题了。