在GPU上运行PyTorch
创始人
2025-01-10 06:06:05
0

文章目录

  • 1、查看GPU的CUDA版本
  • 2、下载CUDA版本
  • 3、安装cuDNN
  • 4、配置CUDA环境变量
  • 5、安装配置Anaconda
  • 6、使用Anaconda
  • 7、pycharm导入虚拟环境
  • 8、安装带GPU的PyTorch⭐
  • 9、总结

🍃作者介绍:双非本科大三网络工程专业在读,阿里云专家博主,专注于Java领域学习,擅长web应用开发、数据结构和算法,初步涉猎人工智能和前端开发。
🦅个人主页:@逐梦苍穹
📕所属专栏:人工智能
🌻gitee地址:xzl的人工智能代码仓库
✈ 您的一键三连,是我创作的最大动力🌹

1、查看GPU的CUDA版本

桌面右键->NVIDIA控制面板->帮助->系统信息->组件:
image.png
image.png
可以看到我这里的CUDA版本是11.8(这里也推荐是11.8,后续会再提到)

2、下载CUDA版本

英伟达官网:https://developer.nvidia.com/cuda-toolkit-archive

到英伟达官网下载对应的CUDA版本,我下载的版本是CUDA11.8.0:
image.png
安装CUDA:双击执行下载的exe文件,会先解压文件到临时目录(不是安装目录),保持默认即可
20200427135351427.png
5. 安装过程:选择自定义
20200427140055597.png
 取消勾选 Visual Studio Integration
20200427140309411.png
 建议默认安装在C盘 。
20200427143306136.png

3、安装cuDNN

英伟达官网:https://developer.nvidia.com/cudnn

到英伟达官网下载与CUDA对应的cuDNN
image.png
下载解压缩之后,将CUDNN文件夹里面的bin、include、lib文件夹里面的文件,直接复制到CUDA的对应的安装目录下 !

4、配置CUDA环境变量

在path中添加如下路径:

C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v11.8\bin
C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v11.8\libnvvp

image.png
在终端输入nvcc -V:
image.png

5、安装配置Anaconda

清华镜像站:https://mirrors.tuna.tsinghua.edu.cn/anaconda/archive/?C=M&O=D

image.png
下载完成之后,正常下一步即可。这里写几个需要注意的点:
①这里可以先不用选Add sys path,只勾选Register。环境变量自己配置。
20210703175041944.png
最后两项都不需要选,点击Finish:
20210703175555398.png
可以看到安装后,除了Anaconda,默认还带了Jupyter、Spyder等:
image.png
image.png
配置环境变量:
image.png
如果你是默认安装的,那你的配置路径就是:
20210703180523269.png
检查是否配置成功:
image.png
在cmd配置国内镜像源:

conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/free/
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main/
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/conda-forge/

6、使用Anaconda

打开Anaconda Navigator:
image.png
启动完成:
image.png
这里是管理环境的:
image.png
conda常用命令:
https://docs.conda.io/projects/conda/en/latest/commands/index.html

7、pycharm导入虚拟环境

image.png

8、安装带GPU的PyTorch⭐

在cmd中(记得用管理员身份打开cmd):

  1. conda install pytorch torchvision torchaudio pytorch-cuda=11.8 -c pytorch -c nvidia
  2. conda install pytorch torchvision torchaudio cudatoolkit=11.8 -c pytorch

测试安装是否成功以及能否正常使用GPU:

print("PyTorch版本: ", torch.__version__)  # 打印PyTorch版本 print("torchvision版本 ", torchvision.__version__)  # 打印torchvision版本 print("CUDA是否可用: ", torch.cuda.is_available())  # 检查CUDA是否可用 

image.png
检查自己电脑的CUDA设备:

# -*- coding: utf-8 -*- # @Author: CSDN@逐梦苍穹 # @Time: 2024/7/16 7:51 import torch  if torch.cuda.is_available():     num_cuda_devices = torch.cuda.device_count()     print(f"Number of CUDA devices: {num_cuda_devices}")     for i in range(num_cuda_devices):         print(f"CUDA Device {i}: {torch.cuda.get_device_name(i)}") else:     print("No CUDA devices available.") 

image.png

9、总结

此次问题折腾了很久,其中的关键就是,电脑的CUDA版本和pytorch-cuda、cudatoolkit的版本必须一致!

相关内容

热门资讯

一分钟内幕!科乐吉林麻将系统发... 一分钟内幕!科乐吉林麻将系统发牌规律,福建大玩家确实真的是有挂,技巧教程(有挂ai代打);所有人都在...
一分钟揭秘!微扑克辅助软件(透... 一分钟揭秘!微扑克辅助软件(透视辅助)确实是有挂(2024已更新)(哔哩哔哩);1、用户打开应用后不...
五分钟发现!广东雀神麻雀怎么赢... 五分钟发现!广东雀神麻雀怎么赢,朋朋棋牌都是是真的有挂,高科技教程(有挂方法)1、广东雀神麻雀怎么赢...
每日必看!人皇大厅吗(透明挂)... 每日必看!人皇大厅吗(透明挂)好像存在有挂(2026已更新)(哔哩哔哩);人皇大厅吗辅助器中分为三种...
重大科普!新华棋牌有挂吗(透视... 重大科普!新华棋牌有挂吗(透视)一直是有挂(2021已更新)(哔哩哔哩)1、完成新华棋牌有挂吗的残局...
二分钟内幕!微信小程序途游辅助... 二分钟内幕!微信小程序途游辅助器,掌中乐游戏中心其实存在有挂,微扑克教程(有挂规律)二分钟内幕!微信...
科技揭秘!jj斗地主系统控牌吗... 科技揭秘!jj斗地主系统控牌吗(透视)本来真的是有挂(2025已更新)(哔哩哔哩)1、科技揭秘!jj...
1分钟普及!哈灵麻将攻略小,微... 1分钟普及!哈灵麻将攻略小,微信小程序十三张好像存在有挂,规律教程(有挂技巧)哈灵麻将攻略小是一种具...
9分钟教程!科乐麻将有挂吗,传... 9分钟教程!科乐麻将有挂吗,传送屋高防版辅助(总是存在有挂)1、完成传送屋高防版辅助透视辅助安装,帮...
每日必看教程!兴动游戏辅助器下... 每日必看教程!兴动游戏辅助器下载(辅助)真是真的有挂(2025已更新)(哔哩哔哩)1、打开软件启动之...