深度学习环境搭建anaconda+pytorch+gpu

一、升级nvrdia 显卡驱动

1、安装cuda

2、安装cudnn

二、安装anaconda

1、官网下载对应版本：https://www.anaconda.com/products/individual

打开anaconda prompt 终端，执行conda list 查看详细信息

2、安装虚拟环境

conda create -n environment_name python=X.X

默认安装到c盘，建议安装到空间较大的盘

conda create --prefix=G:\project\deeplearn\envs\scanner python=3.6

可以查看相关得到帮助文档

conda create --help

3、激活环境

activate G:\project\deeplearn\envs\scanner

4、退出虚拟环境（按需执行）

deactivate

5、更新anaconda 的源为国内的(这步比较重要，我更新清华源的时候下载依然很卡，意外发现了外国语的源非常好用)

c：/user/用户名/.condarc这个文件添加以下内容，然后执行 conda clean -i

channels:
  - https://mirrors.bfsu.edu.cn/anaconda/cloud/pytorch/
  - defaults
show_channel_urls: true
default_channels:
  - https://mirrors.bfsu.edu.cn/anaconda/pkgs/main
  - https://mirrors.bfsu.edu.cn/anaconda/pkgs/r
  - https://mirrors.bfsu.edu.cn/anaconda/pkgs/msys2
custom_channels:
  conda-forge: https://mirrors.bfsu.edu.cn/anaconda/cloud
  msys2: https://mirrors.bfsu.edu.cn/anaconda/cloud
  bioconda: https://mirrors.bfsu.edu.cn/anaconda/cloud
  menpo: https://mirrors.bfsu.edu.cn/anaconda/cloud
  pytorch: https://mirrors.bfsu.edu.cn/anaconda/cloud/
  simpleitk: https://mirrors.bfsu.edu.cn/anaconda/cloud

 

然后缺什么模块继续安装就好了，如果这些源中不包含，那就添加-c使用默认的国外源下载就好了

例如：

conda install -c conda-forge tqdm

6、常用指令

conda list  ：查看已安装的package信息。

conda config --show ：查看conda 配置的详细信息。

三、安装pytorch

进入官网https://pytorch.org/

选择对应的版本，例如最新版本

pytorch

 

conda install pytorch torchvision torchaudio cudatoolkit=10.2

执行以上指令即可。