pytorch安装教程(GPU：Nvidia GTX 3060)

清疚 2023-10-14 10:26 9阅读 0赞

最近进行深度学习，需要要装一个pytorch，需要使用到GPU，网上找了很多教程虽然可以成功安装但是最后GPU显示无法使用，这篇文章就这个情况进行一下说明：

使用GPU版的pytorch需要根据显卡的型号，选择合适的pytorch版本和cuda版本，我这篇文章以3060为例了，如果不一样的话还需要自己调整

### 一、安装Anaconda环境 ###

可以参考这篇文章的第一部分进行安装：[https://blog.csdn.net/qq\_41856733/article/details/127498268][https_blog.csdn.net_qq_41856733_article_details_127498268]

### 二、创建虚拟环境 ###

// 创建虚拟环境pytorch python版本为3.8（这里选择3.8的原因是，我自己亲测3.8是适配的，别的没试过）
    conda create -n pytorch python=3.8

创建完成后进入虚拟环境

conda activate pytorch

### 三、安装pytorch ###

在安装之前要确认下自己的3060支持情况，在不同时间也不一样：在cmd中输入：

nvidia-smi

查看cuda最大支持版本（一般是向下兼容）：  
![在这里插入图片描述][681e9ec08437433ba69f2f814e4bdb32.png]  
下面去官网：[https://pytorch.org/get-started/previous-versions/][https_pytorch.org_get-started_previous-versions]找合适的版本(我选择了1.13.1的pip版本下载)：  
![在这里插入图片描述][7ad2350d53fa4efbbe702a4393666b28.png]  
下载命令是：

pip install torch==1.13.0+cu117 torchvision==0.14.0+cu117 torchaudio==0.13.0 --extra-index-url https://download.pytorch.org/whl/cu117

等待安装成功

### 四、检验GPU版本的pytorch是成功安装 ###

进入python环境后输入：

import torch
    torch.cuda.is_available()

如果结果为True说明安装成功：  
![在这里插入图片描述][84eb2a6e8efc42438dccebbf24e0bba1.png]  
如果还是为False的话，可以检查下上面的步骤是否正确~

[https_blog.csdn.net_qq_41856733_article_details_127498268]: https://blog.csdn.net/qq_41856733/article/details/127498268
[681e9ec08437433ba69f2f814e4bdb32.png]: https://img-blog.csdnimg.cn/681e9ec08437433ba69f2f814e4bdb32.png
[https_pytorch.org_get-started_previous-versions]: https://pytorch.org/get-started/previous-versions/
[7ad2350d53fa4efbbe702a4393666b28.png]: https://img-blog.csdnimg.cn/7ad2350d53fa4efbbe702a4393666b28.png
[84eb2a6e8efc42438dccebbf24e0bba1.png]: https://img-blog.csdnimg.cn/84eb2a6e8efc42438dccebbf24e0bba1.png