pytorch环境配置（anaconda+duda+cudnn+pytorch)

目录

一：pytorch官网一键法

二：自行离线配置法

一：pytorch官网一键法

1、准备工作：

下载anaconda，后续会用到其中的jupyter notebook和anaconda prompt。

2、创建环境：

原因：为了避免冲突及容纳多版本pytorch，所以在base的基础上创建需要的python版本环境。

操作：

（1）打开anaconda prompt，在base环境中创建 conda create -n environment_name（自定义） python=3.7。等待创建完毕。

（2）进入创建的环境 输入conda activate environment_name

3、安装pytorch：

pytorch安装非常的人性化，进入pytorch官网，点击install，选择所需要的选项：

对于每一项的选择在这里做一下详细介绍：

① Pytorch Build：建议选Stable稳定版

② Your OS：根据你自己系统选择，这里我选Windows

③ Package：因为我们已经安装Anaconda，所以在Anaconda Prompt环境下我们可以直接使用Conda方式进行安装，所以选Conda

④ Language：Python

⑤ CUDA：关于GPU的CUDA版本问题如何选择：

第一种方式：打开cmd，输入nvidia-smi，可以看到CUDA的版本信息（没有英伟达显卡，或者显卡不支持CUDA或者就想用CPU运行的话直接选None就行）若此方法不行，可选择以下方法。

第二种方法：进入自己电脑的NVIDIA设置，点击帮助-系统信息，查看自己的版本信息。再与下表对应选择即可。如版本为451.67，则可选择10.1,10.2,11.0等。

若以上方法实在太慢，可以参考以下方法。

二：自行离线配置法

1、准备工作：

下载anaconda，后续会用到其中的jupyter notebook和anaconda prompt。

2、创建环境：

原因：为了避免冲突及容纳多版本pytorch，所以在base的基础上创建需要的python版本环境。

操作：

（1）打开anaconda prompt，在base环境中创建 conda create -n environment_name（自定义） python=3.7。等待创建完毕。

（2）进入创建的环境 输入conda activate environment_name

3、下载安装cuda 

python虚拟环境创建、cuda环境配置和pytorch1.7.1安装_哔哩哔哩_bilibili

4、下载安装cudnn

见以上链接

5、下载安装pytorch

见以上链接

也可以参考这个链接：安装PyTorch GPU/CPU 两种方法，各种版本，稳稳的！_哔哩哔哩_bilibili

注：

1、验证是否安装成功：

conda list指令查找一下有没有pytorch

输入 python，进入 python 环境，输入 import pytorch，如果没有报错，说明安装成功。再输入 torch.cuda.is_available() 检查 GPU 是否可以被 pytorch 使用。

2、换源安装：

conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/free/
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main/
# 以上两条是Anaconda官方库的镜像

conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/conda-forge/
# 以上是Anaconda第三方库 Conda Forge的镜像

# for linux
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/pytorch/
# for legacy win-64
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/peterjc123/
以上两条是Pytorch的Anaconda第三方镜像

conda config --set show_channel_urls yes

conda config –add channelshttps://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/free/
conda config –add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main/
conda config –set show_channel_urls yes
conda config –add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/pytorch/
conda install pytorch torchvision cudatoolkit=10.1（这里是pytorch官网的命令去掉末尾的-c pytorch,去掉之后才是用清华源下载)