如何安装pytorch

https://blog.csdn.net/yuanzhoulvpi/article/details/86742729
https://blog.csdn.net/qq_34342852/article/details/97673819

用conda安装会比较方便,下面直接用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/pkgs/free/
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main/
conda config --append channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/fastai/
conda config --append channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/pytorch/
conda config --append channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/bioconda/

注意:没有自己电脑GPU,就不要安装CUDA了

然后,用这个命令安装:

conda install pytorch torchvision cudatoolkit=10.2    
# 注意,cudatoolkit=10.2这个用官网提供给你的命令,还有后面的 -c pytorch 删掉,因为这里是用清华的镜像安装
原文地址:https://www.cnblogs.com/douzujun/p/13055874.html