1、anaconda安装时间很长,莫名原因,我以为卡了,结果放了两个小时安好了。
2、cuda有可能测不到你的VS,你就勾住勾选,next就行,然后最后会下载失败VS,其他安装成功就行。
查看cuda版本方式,桌面右键选NVIDIA控制面板——>左下角系统信息——>
然后去cuda网站找到你这个对应的版本安装就行,下载差不多2G,有点慢。
3、搞深度学习要cuDNN,这个不安装,你就搜一下,官网注册(qq邮箱就行,不复杂),下载就行,版本啥的,官网页面显示了,你和cuda对应就行。下完里面的三个文件夹复制到cuda的下载目录,就完事。
4、然后是pytorch安装。
先搞个虚拟环境,打开anaconda prompt,进去的是基本环境。
conda create -n py3.7 python=3.7
上面就是创建了名字py3.7的虚拟环境,python版本号是3.7
然后进去这个环境 activate py3.7
这会就切换了环境了。进去之后安装pytorch,就一行命令。去pytorch官网找到对应你cuda版本的连接,官网都显示了。我这10.2的cuda是下面的命令。
conda install pytorch==1.10.0 torchvision==0.11.0 torchaudio==0.10.0 cudatoolkit=10.2 -c pytorch
安装完完事。这会就装完了,想测试就 命令 python 然后 import torch
没啥报错就没问题。
5、设置PyCharm
菜单栏 “文件” “设置“
按着上面依次选就行,上面两个目录就是安装时候那,你创建环境有个envs,里面找到对用的环境之后,选python.exe 然后主页面输入代码
import torch print(torch.__version__) print(torch.cuda.is_available())