如何安装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
- 然后到官网,下载相应的版本
https://pytorch.org/
注意:没有自己电脑GPU,就不要安装CUDA了
然后,用这个命令安装:
conda install pytorch torchvision cudatoolkit=10.2
# 注意,cudatoolkit=10.2这个用官网提供给你的命令,还有后面的 -c pytorch 删掉,因为这里是用清华的镜像安装