安装与卸载pytorch

安装pytorch 方法一
运行以下命令,可将清华镜像添加至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 config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/pytorch/

进入pytorch官网,选择你需要的OS、Package、Language和CUDA,得到下载命令。
安装与卸载pytorch
文章图片

pytorch官网给出的下载命令为(使用conda下载):
conda install pytorch torchvision torchaudio cudatoolkit=9.2 -c pytorch

实际操作发现,添加镜像源之后,安装pytorch依然花费很长的时间,这是因为“ -c pytorch ”。
-c pytorch 指定了conda获取pytorch的channel,指定为conda自带的pytorch仓库,下载会比较慢。
因此,只需将 -c pytorch 去掉,就可以使用清华镜像源快速安装pytorch了。
pip install torch==1.7.0+cu92 torchvision==0.8.1+cu92 torchaudio==0.7.0 -f https://download.pytorch.org/whl/torch_stable.html

同理,如果使用pip下载,去掉pip命令中的 “ -f https://download.pytorch.org/... ”
安装pytorch 方法二(推荐)
在 https://download.pytorch.org/whl/cu90/torch_stable.html 中找到对应版本,下载下来;然后到whl文件下载的位置,使用 pip install xxx.whl 进行安装即可。
安装torchvision
【安装与卸载pytorch】注意:torchvision需要安装0.2.1系列的,否则会报错!!!
pytorch0.4.1 与 torchvision0.2.1 对应
命令:
pip install torchvision==0.2.1

查看当前pytorch版本
import torch print(torch.__version__)

卸载pytorch
# 使用conda卸载Pytorch conda uninstall pytorch conda uninstall libtorch# 使用pip卸载Pytorch pip uninstall torch

    推荐阅读