pytorch安装及卸载
一、pytorch安装
我是已经安装过了gpu版本的tensorflow
1.1 先打开cmd看看自己安装的cuda版本
输入
nvcc --version
可以看到是cuda 10.0，当然你也可以更新英伟达驱动，重新安装更高版本的cuda，使用更高版本的pytorch
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版本
或者从https://pytorch.org/这里看，然后下载离线包，
gpu pytorch对应的torchvision版本也得是GPU的。
; 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命令安装的，则用第二种方法。或者两种方法都试一下。
Original: https://blog.csdn.net/mao_hui_fei/article/details/112078113
Author: 翟羽嚄
Title: pytorch安装及卸载
原创文章受到原创版权保护。转载请注明出处：https://www.johngo689.com/709382/
转载文章受原作者版权保护。转载请注明原作者出处！