如何安装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/

或者,直接改 .condarc 文件

channels:
  - https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main/
  - https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/free/
  - https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/conda-forge/
  - defaults
  - https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/fastai/
  - https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/pytorch/
  - https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/bioconda/
show_channel_urls: yes
remote_connect_timeout_secs: 50.0
remote_read_timeout_secs: 600.0
ssl_verify: true

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

然后,用这个命令安装:

conda install pytorch torchvision cudatoolkit=10.2    
# 注意,cudatoolkit=10.2这个用官网提供给你的命令,还有后面的 -c pytorch 删掉,因为这里是用清华的镜像安装
posted @ 2020-06-06 18:18  douzujun  阅读(605)  评论(0编辑  收藏  举报