安装与卸载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_第1张图片

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

注意: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

你可能感兴趣的