windows10 深度学习训练环境配置

一、安装概要

本说明针对windows安装pytorch GPU训练环境使用（以下软件版本号可以选择其它版本但整体要一致比如cuda10.2，pytorch也要10.2）

二、基本软件安装下载：

1.下载并安装cuda toolkit10.2：

https://developer.download.nvidia.com/compute/cuda/10.2/Prod/local_installers/cuda_10.2.89_441.22_win10.exe

下载并安装cudnn（cuda10.2版本）（需要注册登陆nvidia社区）

https://developer.nvidia.com/rdp/cudnn-download：

https://developer.nvidia.com/compute/machine-learning/cudnn/secure/8.1.0.77/10.2_20210127/cudnn-10.2-windows10-x64-v8.1.0.77.zip

2.下载并安装Anaconda ：

Python 3.8

https://www.anaconda.com/products/individual：

https://repo.anaconda.com/archive/Anaconda3-2020.11-Windows-x86_64.exe

3.下载并安装Pycharm：

https://www.jetbrains.com/pycharm/download/#section=windows

4. 安装Pytorch 1.7.1：

以下a,b两种方法任选一种（注意需要网络）

a.选用conda 安装（傻瓜式安装，可不需要安装第1步cuda及cudnn）

conda install pytorch torchvision torchaudio cudatoolkit=10.2 -c pytorch（官方源安装）

或conda install pytorch torchvision torchaudio cudatoolkit=10.2（不使用官方源）

附conda 添加清华源方法

在C盘用户目录下新建.condarc文件，内容如下：

channels:

  - http://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/pytorch/

  - http://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main/

  - http://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/free/

  - defaults

show_channel_urls: true

b. 选用pip安装（需要配合第1步cuda安装使用）

pip install torch torchvision

附Pip添加清华源方法（或豆瓣源）

在C盘用户目录下新建pip文件夹，在pip文件夹中新建pip.ini文件，文件内容如下：

[global]

index-url = http://pypi.tuna.tsinghua.edu.cn/simple

[install]

trusted-host=pypi.tuna.tsinghua.edu.cn

三、以上安装完基本环境已经配置完成，但训练时仍然需要安装其他python依赖库（如opencv-python，onnx等等，可使用pip install 安装）