由于是初学者,文章主要目的是记录学习过程,可能不是非常完备。
这里是在anaconda中建立了“CNN_E”的新环境,问题出在之前安装的CUDA版本为11.2。
但离谱的是pytorch官网没有对应版本。
在别的文章的建议下,尝试下载pytorch==1.10.0 对应cudatookit=11.3。(指令如下)
(图1)
但是运行不成功,报错如下图
(图2)
问题可能出现在环境中python的版本,不兼容or ...?(感觉有时候很玄学)
总之看报错意思可能是python版本有点高。于是新建了一个环境,测试是不是猜想的问题。(如下图)
(图3)
将python版本降到3.8。
(图4)
依然运行‘图1’安装程序,成功解决问题。
(图5)
经检验也能够使用电脑GPU。(ps:演示电脑gpu为3060,不知道gpu不一样有没有影响)
(图6)
=============================END==============================