win10和ubuntu18.04下用anoconda新建虚拟环境安装pytorch

记录一下分别在win10环境和ubuntu18.04环境下用anoconda安装pytorch的过程：

一、win10环境：

1. 打开anoconda prompt：

win+q打开搜索框，搜索anoconda prompt并打开。

2. 查看环境：

用以下命令查看所有环境，*表示为当前所在环境，默认为root。

conda env list
# 或
conda info -e

这里之前装过一个pytorch环境，用删除，重新装一遍熟悉下过程。

conda remove -n pytorch --all

3. 配置清华TUNA镜像：

# 添加Anaconda的TUNA镜像
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/free/
# 设置搜索时显示通道地址
conda config --set show_channel_urls yes

建议参考这篇博客的内容https://blog.csdn.net/watermelon1123/article/details/88122020，添加第三方conda源以提高速度。

这里直接把配置信息贴出来，可以直接复制粘贴到C:\Users\admin\.condarc文件中：

channels:
  - https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/bioconda/
  - https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/pytorch/
  - https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/menpo/
  - https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/msys2/
  - https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/conda-forge/
  - https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main/
  - https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/free/
show_channel_urls: true
ssl_verify: true

4. 创建虚拟环境：

命令为：

conda create -n pytorch python=3.6 anaconda

（其中加入最后一项anaconda命令建立虚拟环境时，会同时安装例如Notebook、Numpy、Scipy、Matplotlib、Pandas等数据分析包，用户也可以不加，选择后续安装。）

完成后conda env list查看，虚拟环境已经建好了。

 5. 激活环境：

activate pytorch

激活创建的虚拟环境，conda list可查看安装的包。

6. 安装pytorch:

首先打开NVIDIA 控制面板，点击 帮助→系统信息→组件 查看CUDA版本，然后到官网选择对应的CUDA版本，复制并运行生成的命令。（如果没有对应的CUDA版本，到previous-versions里找）

运行的命令要去掉后面的 -c pytorch，否则就会指向conda官方的pytorch下载库通道，比如我运行的命令为:

conda install pytorch torchvision cudatoolkit=10.0

 

如果你的网络总断开的话，建议去镜像网站将文件下载下来，用以下命令本地安装。

conda install --use-local your-pkg-name

7. 测试

二、ubuntu18.04环境:

1. 新建虚拟环境：

conda create -n pytorch python=3.6

2. 激活环境：

source activate pytorch

3. 修改.condarc文件，与window下文件一样添加清华镜像源：

vim ~/.condarc

修改后wq保存退出

4. 查看cuda版本，我的是10.0：

cat /usr/local/cuda/version.txt

顺便写一下查看cudnn的方法：

cat /usr/local/cuda/include/cudnn.h | grep CUDNN_MAJOR -A 2

5. 跟window下一样，去官网复制并运行对应版本的命令：

conda install pytorch==1.2.0 torchvision==0.4.0 cudatoolkit=10.0

6. 测试：

参考：

https://blog.csdn.net/watermelon1123/article/details/88122020

https://www.cnblogs.com/carlyao/p/6929459.html

https://www.jianshu.com/p/ee08ca2c1160