github: https://github.com/peterjc123/pytorch-scripts
首先要安装anaconda:
推荐清华镜像
https://mirrors.tuna.tsinghua.edu.cn/anaconda/archive/
拖到最后,下载最新版即可。
https://blog.youkuaiyun.com/watermelon1123/article/details/88122020 </div>
比较常见的方法是通过pip和conda安装,当使用conda安装的时候,可能会遇到下载过慢的问题,尤其是文件:pytorch-1.0.1-py3.6_cuda90_cudnn7_1.tar.bz2,下载的时候可能会遇到无尽的等待。这里推荐用清华源替代默认conda源的方法,可以解决下载过慢的问题。
清华conda源地址:https://mirrors.tuna.tsinghua.edu.cn/help/anaconda/
添加清华源
-
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源:
-
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/cloud/msys2/
-
conda config --
add channels https:
//mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/bioconda/
-
conda config --
add channels https:
//mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/menpo/
-
conda config --
add channels https:
//mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/pytorch/
一行代码:
sudo ~/anaconda3/bin/pip install torch torchvision
制定torch版本用下面一行安装
sudo ~/anaconda3/bin/pip install torch1.0.1 torchvision0.2.2