新手小白配置教程windows深度学习环境：anaconda+pycharm+cuda+cudnn+pytorch

目录

前言

下载并安装anaconda

下载并安装pycharm 

更新NVIDIA驱动 

安装cuda

安装cudnn

安装pytorch

验证pytorch

---

前言

新手小白配置windows深度学习环境：anaconda+pycharm+cuda+cudnn+pytorch，重装过几次系统，这次把配置过程记录下来，以后直接按这个流程无脑安装了

软硬件配置：

windows11 家庭中文版 21H2 22000.856

NVIDIA GeForce RTX 3070 Ti Laptop GPU

anaconda 2022.05

pycharm professional 2022.2.1

NVIDIA驱动 516.94

cuda 11.6.2

cudnn 8.5

pytorch 1.12.1 及对应torchvision、torchaudio

---

下载并安装anaconda

进入anaconda官网Anaconda | The World's Most Popular Data Science Platform下载anaconda。

 

---

下载并安装pycharm 

 进入pycharm官网PyCharm: the Python IDE for Professional Developers by JetBrains下载专业版。

添加环境变量一定要勾上，不然还要手动添加环境变量。

---

更新NVIDIA驱动 

建议无论安装与否都可以更新下驱动，打开任务管理器查看GPU型号

官方驱动 | NVIDIA下载并更新NVIDIA驱动，选择对应型号

下载好后打开 

 点击下一步等待安装完成。 

---

安装cuda

按win+r输入cmd进入命令行

输入nvidia-smi查看可安装的cuda版本号

版本是向下兼容的，11.7表示最高支持11.7，可安装11.7及以下版本

打开pytorch下载地址：PyTorch

显示目前只到11.6的cuda，所以为了无脑安装，cuda就下载11.6版本的

cuda历史版本下载地址：CUDA Toolkit Archive | NVIDIA Developer

 

 

有人说要配置cuda环境变量，我没有配置，也没遇到过这方面的错误，等遇到了再补充。

---

安装cudnn

 下载地址：CUDA Deep Neural Network (cuDNN) | NVIDIA Developer

 需要注册登录

下载解压后，将cudnn-windows-xxxxxxxxx下三个文件夹里的文件分别复制到cuda目录相应文件夹中

有人也说要配置cudnn环境变量，我没有配置，也没遇到过这方面的错误，等遇到了再补充。

---

安装pytorch

打开anaconda prompt

建议新建环境 不要在base中安装pytorch

输入conda create -n pytorch python=3.9创建名为pytorch的环境

此时输入conda env list会出现pytorch环境

输入conda activate pytorch进入pytorch环境

进入pytorch官网地址：PyTorch

 

 复制命令到anaconda prompt安装pytorch torchvision torchaudio

 安装成功

---

验证pytorch

打开pycharm并激活

新建项目

创建好后输入以下代码测试pytorch是否安装成功

import torch
print(torch.__version__)

输出：

1.12.1+cu116

安装成功

若有代码需要pytorch环境，打开代码项目后，选择File->Settings

选择pytorch环境