如何快速安装PyTorch 1.4.1 GPU版

1、创建虚拟环境

打开anaconda操作窗口，输入如下命令：

conda create -n  torch14 python=3.6

torch14为虚拟环境名称

conda activate torch14即可进入创建的虚拟环境中。

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/
conda config --set show_channel_urls yes
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/pytorch/

3. 安装 pytorch

conda install pytorch==1.4.0 torchvision==0.5.0 cudatoolkit=10.1

请注意，此处无“ -c pytorch”，“ -c pytorch”为从官网下载，速度简直不忍直视。

4、等待片刻即可安装成功。