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

  最近进行深度学习,需要要装一个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

创建完成后进入虚拟环境

conda activate pytorch

三、安装pytorch

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

nvidia-smi

查看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

等待安装成功

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

进入python环境后输入:

import torch
torch.cuda.is_available()

如果结果为True说明安装成功:
在这里插入图片描述
如果还是为False的话,可以检查下上面的步骤是否正确~

猜你喜欢

转载自blog.csdn.net/gls_nuaa/article/details/132020230