pytorch安装及卸载

pytorch安装及卸载

一、pytorch安装

我是已经安装过了gpu版本的tensorflow

1.1 先打开cmd看看自己安装的cuda版本

输入

nvcc --version

可以看到是cuda 10.0

1.2 找到离线下载包进行安装

pytorch离线下载地址：pytorch离线下载

根据windows系统+python版本——cuda版本进行下载

1.3 使用pip安装

pip install torch-1.0.0-cp36-cp36m-win_amd64.whl

torch 1.0.0 对应的torchvision的版本为0.2.0

pip install torchvision==0.2.0

其他对应的版本：
torch对应的torchvision版本

1.4 验证gpu版本是否安装成功

检测程序如下：

import torch

print(torch.__version__)
print(torch.cuda.is_available())

运行结果：

1.0.0
True

PyTorch版本为1.0.0
True代表GPU版本成功安装！

二、pytorch卸载

1、使用conda卸载Pytorch

conda uninstall pytorch
conda uninstall libtorch

2、使用pip卸载Pytorch

pip uninstall torch

如果当初是用conda install 命令安装的pytorch，则用第一种方法，如果是用pip install命令安装的，则用第二种方法。或者两种方法都试一下。