RTX 3080 安装 pytorch-gpu 1.12.0 + torchvision 0.13.0 + CUDA 11.3.0 + cudnn 8.8 .0

最新推荐文章于 2025-04-03 21:49:19 发布

Despereaux_

最新推荐文章于 2025-04-03 21:49:19 发布

阅读量6.4k

点赞数 7

文章标签： pytorch python 深度学习

本文链接：https://blog.csdn.net/Despereaux_/article/details/129322034

版权

文章描述了在conda环境下安装PyTorchGPU版本时遇到的问题及解决方法。首先，由于conda默认安装的可能是CPU版本，需要从PyTorch官网下载对应版本的wheel文件手动安装。接着，下载并安装CUDA11.3和cudnn8.8.0，然后将它们复制到CUDA安装目录。最后，通过`torch.cuda.is_available()`检查GPU支持情况，确保安装成功。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

直接conda install安装pytorch gpu遇到的问题

一、使用conda创建虚拟环境

二、下载pytorch gpu版本

三、安装pytorch、torchvision和torchaudio

四、下载CUDA11.3和cudnn8.8.0

五、安装CUDA的流程

六、安装cudnn

七、检查pytorch gpu是否可以运行

直接conda install安装pytorch gpu遇到的问题

一开始我时选择直接在pytorch官网复制命令直接conda安装：

安装完成后运行下列程序显示为False，说明GPU版本没有安装成功。

import torch
torch.cuda.is_available()

输入conda list命令，发现安装的pytorch为cpu版本。这是由于conda安装的时候会按照这四个pytorch、torchvision、torchaudio和cudatoolkit的版本同时在当前的安装包库内匹配，没有发现pytorch gpu版本就会自动用cpu版本代替。

因此，需要在https://download.pytorch.org/whl/torch_stable.html网站下载上述pytorch1.12.0、torchvision0.13.0、torchaudio0.12.0这三个安装包，手动安装。

一、使用conda创建虚拟环境

首先安装Anaconda，使用Anaconda Powershell Prompt创建一个虚拟环境，以便于后续环境的管理。PS：运行别人的项目时最好配置新的环境。

这里创建一个python版本为3.7的虚拟环境，环境名叫”pytorch“：

conda create -n pytorch python=3.7

然后激活这个环境：

conda activate pytorch

此时的环境由base切换到创建的虚拟环境pytorch中

二、下载pytorch gpu版本

打开网站：

https://download.pytorch.org/whl/torch_stable.html

Ctrl+F搜索：cu113/torch-1.12.0

可以看到出现上面几个安装包，cu113表示CUDA1.13，即为GPU版本。torch-1.12.0为pytorch版本。cp表示python版本，之前创建的虚拟环境pyhon=3.7，因此这里选择cp37，然后选择windows系统的win_amd64.

按照此方法，在网页https://download.pytorch.org/whl/torch_stable.html中下载所需的三个安装包：

cu113/torch-1.12.0%2Bcu113-cp37-cp37m-win_amd64.whl

cu113/torchvision-0.13.0%2Bcu113-cp37-cp37m-win_amd64.whl

cu113/torchaudio-0.12.0%2Bcu113-cp37-cp37m-win_amd64.whl

三、安装pytorch、torchvision和torchaudio

在pytorch虚拟环境中，使用pip install 命令安装下载好的三个文件

pip install D:\Download\IEdownload\torch-1.12.0+cu113-cp37-cp37m-win_amd64.whl
pip install D:\Download\IEdownload\torchvision-0.13.0+cu113-cp37-cp37m-win_amd64.whl
pip install D:\Download\IEdownload\torchaudio-0.12.0+cu113-cp37-cp37m-win_amd64.whl

pip install后面为文件下载的路径