1、添加国内镜像
参考:conda安装Pytorch下载过慢解决办法
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
2、安装pytorch
进入pytorch官网选择操作系统和版本,下面就能给出推荐的安装命令,建议使用anaconda安装
# 注意不要按照官网的提示在最后加 -c pytorch,去掉后才会从添加的镜像源下载
conda install pytorch torchvision cuda80
# 如果需要安装指定版本的pytorch,参照:
# https://pytorch.org/get-started/previous-versions/ 例如:
conda install pytorch=0.4.0 torchvision cuda80
3、验证是否安装成功
source activate your_conda_env # 或者是conda activate,谁能告诉我二者的区别。。。
python
>>>import torch
>>>print(torch.__version__)
若安装成功,会输出当前pytorch的版本号
根据笔者经验,大多安装不成功的原因均来自cuda版本,设置好安装的cudatoolkit版本即可