pytorch安装及卸载

清疚 2023-01-01 09:54 772阅读 0赞

## pytorch安装及卸载 ##

# 一、pytorch安装 #

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

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

输入

nvcc --version

![在这里插入图片描述][20210101230837100.png]  
可以看到是cuda 10.0，当然你也可以更新英伟达驱动，重新安装更高版本的cuda，使用更高版本的pytorch

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

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

根据windows系统+python版本——cuda版本进行下载  
![在这里插入图片描述][watermark_type_ZmFuZ3poZW5naGVpdGk_shadow_10_text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L21hb19odWlfZmVp_size_16_color_FFFFFF_t_70]

## 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版本][torch_torchvision]

或者从[https://pytorch.org/][https_pytorch.org]这里看，然后下载离线包，  
![在这里插入图片描述][watermark_type_ZmFuZ3poZW5naGVpdGk_shadow_10_text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L21hb19odWlfZmVp_size_16_color_FFFFFF_t_70 1]  
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命令安装的，则用第二种方法。或者两种方法都试一下。

[20210101230837100.png]: /images/20221120/e7703b8b263240ddb25daa21321124dd.png
[pytorch]: https://download.pytorch.org/whl/torch_stable.html
[watermark_type_ZmFuZ3poZW5naGVpdGk_shadow_10_text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L21hb19odWlfZmVp_size_16_color_FFFFFF_t_70]: /images/20221120/2c043c251679482f9e5f82da660f07ee.png
[torch_torchvision]: https://blog.csdn.net/studyvcmfc/article/details/109404951
[https_pytorch.org]: https://pytorch.org/
[watermark_type_ZmFuZ3poZW5naGVpdGk_shadow_10_text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L21hb19odWlfZmVp_size_16_color_FFFFFF_t_70 1]: https://img-blog.csdnimg.cn/d536910b818b4c7aa7150d382ff27118.png?x-oss-process=image/watermark,type_ZmFuZ3poZW5naGVpdGk,shadow_10,text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L21hb19odWlfZmVp,size_16,color_FFFFFF,t_70