具体主要看
https://blog.csdn.net/m0_37867091/article/details/105788637
很详细的步骤!
1、安装CUDA的不同
但是我的电脑是CUDA 11.2.162的 (如下)
所以在: 英伟达官网链接下载的也就选择了不同的对应版本:
2、安装cuDNN的不同
3. 安装PyTorch的不同!!(格外注意)
由于我在 PyTorch官网、还有csdn中找不到CUDA11.2对应什么版本的pytorch(我只找到了最新的CUDA11.1的对应版本),所以我大胆尝试:
下载CUDA11.1的对应版本!(看到CSDN有小伙伴失败后还是想试一试)
详细如下图:
NOTE部分不用理会(就是提醒你命令后面不要省略了它的’conda-forge’)
也就是在Anaconda Prompt 中输入如下:
conda install pytorch torchvision torchaudio cudatoolkit=11.1 -c pytorch -c conda-forge
几行代码之后,会出现:
输入y同意
然后我按照开头给出的参考教程链接的后面步骤继续试验,总是no moudule name torch错误,我就重新输入刚才说的PyTorch官方给的命令了三四次:
然后终于通过了!证明CUDA11.1是可以用的!!!!
我在过程中还犯了个错误,由于没有安装环境之类的经验,我当时输入的:
没错,我打的是一个下划线,正确的是两个下划线!