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

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

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

一、安装Anaconda环境

可以参考这篇文章的第一部分进行安装：https://blog.csdn.net/qq_41856733/article/details/127498268

二、创建虚拟环境

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

创建完成后进入虚拟环境

conda activate pytorch
1

三、安装pytorch

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

nvidia-smi
1

查看cuda最大支持版本（一般是向下兼容）：

下面去官网：https://pytorch.org/get-started/previous-versions/找合适的版本(我选择了1.13.1的pip版本下载)：

下载命令是：

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
1

等待安装成功

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

进入python环境后输入：

import torch
torch.cuda.is_available()
1
2

如果结果为True说明安装成功：

如果还是为False的话，可以检查下上面的步骤是否正确~