linux下pytorch环境配置

Ubuntu下pytorch的安装

记录一下我配置环境入的各种坑，给你们点参考。第一次配环境配了好几天，一直显示下载失败。

Ubuntu创建一个环境

1. 先创建Python环境 (其中环境名为ww1）：

   conda create –n ww1 python ==3.7

2. 进入该环境：conda activate ww1
   (切换到该环境：source activate ww1)

3. 退出该环境： conda deactivate

在ww1环境下安装pytorch

**pytorch*8版本和你的cuda版本以及python需要对应，这个是非常重要的

一、下面先介绍一些如何查看对应版本
1.清华源

• 清华源版本下包地址：
  https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/

一般情况下使用以下这一个就可以

• pytorch下离线包地址：(Ctrl+F在该界面搜索关键字）
• https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/pytorch/linux-64/

2.pytorch官网查看

• https://pytorch.org/get-started/previous-versions/
  

二、默认情况下下载会很慢

• CUDA 10.1

conda install pytorch=1.6.0 torchvision=0.7.0 cudatoolkit=10.1 -c pytorch

可以使用以下三种方法加快安装速度：

1.将anconda的安装源修改为清华的镜像源：

conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/msys2/
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/conda-forge/
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/free/
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/pytorch/
conda config --set show_channel_urls yes

2.使用pip安装

pip install torchvision==0.7.0 -i https://pypi.tuna.tsinghua.edu.cn/simple
pip install torch==1.6.0 -i https://pypi.tuna.tsinghua.edu.cn/simple

3.离线安装容易下载超时失败的包

后续补充…