系统:win10 已安装:python3.10&pycharm 显卡:MX450独立显卡
一、(仅独显且安装至GPU需要,其余情形可直接跳过)安装CUDA与CUDNN
1.注意事项
CUDA、CUDNN、以及Anaconda虚拟出的环境中所需要安装的PyTorch之间是有版本上的对应关系的,由于下载CUDNN需要在NVIDIA上注册账户且NVIDIA网站似乎很不稳定,因此这里我的做法是在清华的镜像上下载CUDNN。
另外,通过Nvidia控制面板->帮助->系统信息-> 组件 处查看支持的CUDA版本,假设支持的CUDA版本是11.2,那么就可以安装版本号<=11.2的CUDA。
以下是安装链接:
CUDA(版本10.2):CUDA Toolkit 10.2 Download | NVIDIA Developer
【注意】Installer Type一栏建议选择exe(local)
CUDNN(清华镜像,注意下载的CUDNN应当对应于CUDA10.2,从文件名上即可看到):https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main/win-64/cudnn-7.6.5-cuda10.2_0.tar.bz2
Anaconda(清华镜像,安装最新版本,注意名称顺序并非版本顺序):Index of /anaconda/archive/ | 清华大学开源软件镜像站 | Tsinghua Open Source Mirror
2.安装CUDA
按部就班逐步安装即可,更详细的参考以下链接
1.自定义安装需要取消勾选Visual Studio Integration
2.development,documentations,samples的安装目录不建议修改
3.记得配置环境变量(注意要添加在系统变量中)
3.安装CUDNN
在上述链接:https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main/win-64中下载对应的CUDNN镜像,下载后剪切到C:\用户\对应用户名的文件夹下。
打开cmd,输入以下命令:(其中cudnn7.6.5.tar.bz2是文件名,如果不对应重命名一下即可)
conda install --use-local cudnn7.6.5.tar.bz2
然后等待安装即可。
【TIPS】如果在安装CUDA时手动指定了development,documentations,samples的安装目录,那么上述步骤可能无法成功安装,因此development,documentations,samples的安装目录不建议修改。
4.检查CUDA与CUDNN安装是否成功
(1)检查CUDA:在cmd中输入“nvcc -V”,若返回CVDA版本号等信息即可
(2)检查CUDNN:通过NVIDIA提供的 deviceQuery.exe 和 bandwidthTest.exe 来查看GPU的状态,两者均在安装目录的 extras\demo_suite文件夹中,用cmd访问对应文件夹后打开两个.exe文件分别进行检查。(参考:windows 验证CUDA和CUDNN是否安装成功 - 知乎)
二、安装Anaconda
参考以下链接即可。
【Anaconda教程01】怎么安装Anaconda3 - 知乎
【TIPS】
1.同样地,记得配置环境变量(注意要添加在系统变量中)
2.修改清华镜像一步,使用清华镜像帮助中的文件,并将其中所有的https修改为http,链接如下:anaconda | 镜像站使用帮助 | 清华大学开源软件镜像站 | Tsinghua Open Source Mirror。
3.添加环境变量时,注意:
E:\Anaconda\Library\mingw-w64\bin
此条环境变量非必要请勿添加,否则可能对同一电脑上的C++程序编译与调试产生未知干扰(本人是STL中的部分内容无法正常调试)
三、新建虚拟环境与安装PyTorch
参考以下链接即可。
【TIPS】如果安装CPU版本(无NVIDIA独立显卡的只能安装CPU版本),则CUDA一栏选择“CPU”;如果需要安装GPU版本,则CUDA一栏需要与先前安装的CUDA版本一致。
四、其它的一些tips
1.Anaconda安装前,不必卸载已安装的python
在Anaconda安装之后,添加环境变量->path(注意要添加在系统变量里的path,而不是用户变量里的path),只需保证python3.10的两个path在Anaconda的所有path上方,即可实现在cmd中输入python打开python3.10,而在Anaconda Prompt中输入python即可打开Anaconda中base所对应的python(这里我安装的是python3.9)。
这样的操作可以解决在cmd中运行python的以下警告:
Warning: This Python interpreter is in a conda environment
2.卸载Anaconda的办法
如遇安装错误等原因需要彻底卸载Anaconda,可参考以下链接: