1.安装cuda环境
1.1 更新显卡驱动(非必需,只要步骤1.2中的CUDA版本支持11.6及以上就可以)
打开NVIDIA控制面板,可以看到显卡型号:GeForce RTX 2060,显卡驱动版本:512.78
![](https://i-blog.csdnimg.cn/blog_migrate/31464e94430dad68c7a06544860b164f.png)
进入NVIDIA官网检索最新驱动版本并更新(我偷懒没更新):https://www.nvidia.cn/Download/index.aspx?lang=cn
![](https://i-blog.csdnimg.cn/blog_migrate/caa7e5ab7cb1dbed2776356bbc66e30e.png)
1.2 查看显卡对应的CUDA版本
nvidia控制面板->帮助->系统信息->组件,可以看到CUDA版本是11.6.134
![](https://i-blog.csdnimg.cn/blog_migrate/52a101663409ba0145e0b33b4be2a467.png)
![](https://i-blog.csdnimg.cn/blog_migrate/ab2763bb7f88eaa57e6c3db3ee19c9e7.png)
1.3 下载并安装对应版本的CUDA,推荐本地安装(local)
https://developer.nvidia.com/cuda-toolkit-archive
![](https://i-blog.csdnimg.cn/blog_migrate/5aea4c0a98d2d812d53004f40b010940.png)
1.4 下载并安装对应版本的cuDNN,需要注册一个账号
https://developer.nvidia.com/rdp/cudnn-archive
![](https://i-blog.csdnimg.cn/blog_migrate/35912909f09e322363dabb20bac76650.png)
2. 下载并安装anaconda
![](https://i-blog.csdnimg.cn/blog_migrate/5b9f402c46d9a6f160f874c06f489cac.png)
2.1 在conda虚拟环境中安装python
打开Anaconda Powershell Prompt
![](https://i-blog.csdnimg.cn/blog_migrate/6807ec28796be83335a68d0e0a3924b1.png)
输入命令conda create -n env_name python=x.x
conda create -n pytorch python=3.7
……
3. 安装pytorch
方法一:进入官网找到对应版本的安装命令,如果网速很慢的话会需要换源
https://pytorch.org/get-started/locally/
![](https://i-blog.csdnimg.cn/blog_migrate/120f2f0fb4e3a660b53d901bcfac887d.png)
方法二:离线下载安装包到本地,再进行离线安装
https://download.pytorch.org/whl/torch_stable.html
![](https://i-blog.csdnimg.cn/blog_migrate/36ed5c4e57089950c9e4ff506c30b50b.png)
……
4. 安装pycharm专业版,使用学校邮箱或学信网在线报告进行认证免费使用,一年认证一次。配置pycharm项目的解释器。