背景介绍:根据自己要复现的代码的环境要求,我选择了自己的pytorch版本是1.8.1,python==3.7+,再结合自己显卡的cuda版本,比如我是12.4,要选择的cudatoolkit版本应该小于你的cuda,如果找不到合适的,就先去升级你自己显卡的版本,在nVidia升级。
接下来是去pytorch官网找到我需要的pytorch版本的conda命令。
这里我选择了cudatoolkit=11.3,按照此命令去conda安装,最后conda list 检查发现还是cpu版本。
看了很多文章,有的说是用conda里的pip渠道来安装,我试了还是不可以,仍然为cpu版本,这里附上一个链接。好文: