win10系统下使用Anaconda 安装pytorch

1、用conda创建pytorch虚拟环境（在conda prompt环境下完成）

命令：conda create -n environment_name python=X.X

(注：该命令只适用于Windows环境；“environment_name”是要创建的环境名；“python=X.X”是选择的Python版本)

例如我的命令为：conda create -n pytorch python=3.6

2、激活虚拟环境（在conda prompt环境下完成）

命令：activate your_env_name

例如我的命令为：avtivate pytorch

3、安装pytorch

进入官网找到合适的版本：pytorch官网:https://pytorch.org/get-started/locally/

在这里如果不知道CUDA型号的同学，可以通过命令查看一下

（1）打开CMD，输入命令：nvidia-smi

（2）打开CMD，输入命令：nvcc --version

通过查看我的CUDA是10.2版本的。

所以我的指令为：

conda install pytorch torchvision cudatoolkit=10.2 -c pytorch

在这里大家先别急着安装，先换下源。

4、换源

打开anaconda prompt，输入下列指令：

conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/pytorch/
conda config --add channels https://mirrors.sjtug.sjtu.edu.cn/anaconda/pkgs/main/
conda config --add channels https://mirrors.sjtug.sjtu.edu.cn/anaconda/pkgs/free/

5、安装

既然选择了用镜像源下载，那么安装指令的 -c pytorch需删去，也就是：conda install pytorch torchvision cudatoolkit=10.2

 

然后等待安装完毕。后续需要什么包自行安装就好。

（如果能帮助到大家的，点个赞呗~）