ubuntu18.04安装和卸载pytorch

进入官网：Start Locally | PyTorch

选择合适的版本

 

 复制命令安装即可

pip3 install torch torchvision torchaudio --extra-index-url https://download.pytorch.org/whl/cu113

检查：

python
import torch
torch.__version__
print (torch.cuda.is_available());
#ture

a = torch.Tensor(5,3)
print(a)
a.cuda()


import torchvision
torchvision.__version__

卸载命令

pip uninstall torch
pip3 uninstall torchvision

安装包在tmp目录下，可以去手动删除

安装自定义版本的 torch

torch与torchvision版本的对应关系为+1的关系，如下图所示

 指定版本的torch与torchvision安装可参考这篇文章

如何快速安装Pytorch1.10+cuda11.1

 指定版本的torch与torchvision下载地址：

https://download.pytorch.org/whl/torch_stable.html

 下载好后直接使用如下命令安装即可

pip install torch-1.10.2+cu113-cp38-cp38-linux_x86_64.whl 
pip install torchvision-0.11.2+cu113-cp38-cp38-linux_x86_64.whl 
或者  #加上python -m 效果是一样的
python -m pip install torch-1.10.2+cu113-cp38-cp38-linux_x86_64.whl 
python -m pip install torchvision-0.11.3+cu113-cp38-cp38-linux_x86_64.whl