我的笔记本是NVIDIA GeForce RTX 3050 Laptop GPU的,原先安装cuda10.2不能使,查了3050芯片安装哪个版本:
CUDA Toolkit 11.1.1 cuDNN v8.1.1 (Feburary 26th, 2021), for CUDA 11.0,11.1 and 11.2
pytorch==1.8.0 torchvision==0.9.0 torchaudio==0.8.0
参考Windows 11+RTX3050Ti安装CUDA、cuDNN与Pytorch的正确版本-CSDN博客
接下来是具体操作:
先下载安装包:
1.选择CUDA Toolkit 11.1.1
https://developer.nvidia.com/cuda-toolkit-archive
Win-x86-10-exe
2.选择cuDNN v8.1.1 ...... for CUDA 11.0,11.1 and 11.2 ---cuDNN ......Windows(x86)
https://link.zhihu.com/?target=https%3A//developer.nvidia.com/rdp/cudnn-archive
3.下载 torch==1.8.0 torchvision==0.9.0 torchaudio==0.8.0
https://download.pytorch.org/whl/cu111
到现在,准备工作已做完
接下来就是看Python伊甸园老师“如何配置Pytorch的GPU环境”的文章就好
此处省略。。。。。。。
ok挠,现在配置成功(我建的虚拟环境取名TAT)
然后是收尾工作:在pycharm上配置pytorch环境
打开pycharm File->Settings
(ps:这里我一开始没找到我TAT下的python.exe,某办法啊,然后我就重启了)
最后显示-点击0k就好
运行你的项目。。。
就OK了,没有显示userwarning: geforce rtx 3050 laptop gpu with cuda capability sm_86 is not compatible with,成功!
注意:耐心🙋不行就卸了重装,遇到不一样的地方就停一下,看看老师们的具体操作,肯定是能解决的。如果不行,说明你没找到那个“对的”博客🤔多搜搜,问题一定能解决!少啰嗦,干就完了。
再强调:99%来自两位大佬的文章,我只是针对我的设备,做个记录