1.下载pycharm
2.下载anaconda,添加环境变量。路径:此电脑--属性--高级系统设置--环境变量--双击path--新建,添加如下变量。
3.安装cuda和cudnn
查看版本时可能不好查看,可以在找到nvidia-smi这个文件,直接在这个目录里面进入命令行,输入nvidia-smi,可以知道最合适的cuda版本。
进入cuda和cudnn官网下载,将cudnn解压缩包里的这几个文件全部放入对应的cuda安装文件中。相应的,添加好环境变量。nvcc -V 可以查询版本。
C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v11.1\extras\demo_suite(如果自定义了安装路径需要修改该路径)中的 bandwidthTest.exe 和 deviceQuery.exe 文件,如果最后都显示 Result = PASS 则说明cudnn安装成功
4.在promt里面先查看一下环境,随后创建一个虚拟环境
conda env list
conda create -n pytorch python=3.8
创建虚拟环境之前可以添加清华源,在navigator里面直接添加链接即可。
5.pytorch官方还不支持12.2的cuda,这里可以找先验版装,直接在虚拟环境里面安装。
6.在pycharm里面打开虚拟环境时,用虚拟环境的python而不要用anaconda安装目录下的python。找不到python.exe,需要找到conda.bat文件,一般放在Anaconda安装路径下的Library/bin文件下,点击Load Environment,不然会出现找不到第三方库的问题。