Ubuntu 下 conda 加速安装pytorch

1.确定自己的CUDA版本

cat /usr/local/cuda/version.txt

2.建立虚拟环境

conda create -n pytorch

开启、退出、删除命令：

source activate pytorch
source deactivate
conda remove -n pytorch --all

3.安装pytorch

https://pytorch.org/ 在官网主页根据你的系统和CUDA，python版本，选择conda安装方式。

使用官方指定的方法下载太慢，所以：

• 进入anaconda 的清华镜像帮助：网站为：https://mirrors.tuna.tsinghua.edu.cn/help/anaconda/
• 复制以下命令，粘贴回车

conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/pytorch/

• 输入官方提供的安装pytorch的命令（去掉末尾的-c pytorch）
  示例：

 conda install pytorch torchvision cudatoolkit=10.0

4.测试

根据提示安装好了之后可以python然后import torch尝试是否安装成功了。