系统: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,可参考以下链接:

3.在pycharm中使用Anaconda创建的虚拟环境