当前位置: 首页 > news >正文

pytorch配置—什么是CUDA,什么是CUDNN、在配置pytorch虚拟环境中遇到的问题、在安装gpu—pytorch中遇到的问题

1.什么是CUDA,什么是CUDNN

(1)什么是CUDA
CUDA(ComputeUnified Device Architecture),是显卡厂商NVIDIA推出的运算平台。 CUDA是一种由NVIDIA推出的通用并行计算架构,该架构使GPU能够解决复杂的计算问题。
(2)什么是CUDNN
NVIDIA cuDNN是用于深度神经网络的GPU加速库。它强调性能、易用性和低内存开销。NVIDIA cuDNN可以集成到更高级别的机器学习框架中,如谷歌的Tensorflow、加州大学伯克利分校的流行caffe软件。简单的插入式设计可以让开发人员专注于设计和实现神经网络模型,而不是简单调整性能,同时还可以在GPU上实现高性能现代并行计算。
(3)CUDA与CUDNN的关系
CUDA看作是一个工作台,上面配有很多工具,如锤子、螺丝刀等。cuDNN是基于CUDA的深度学习GPU加速库,有了它才能在GPU上完成深度学习的计算。它就相当于工作的工具,比如它就是个扳手。但是CUDA这个工作台买来的时候,并没有送扳手。
想要在CUDA上运行深度神经网络,就要安装cuDNN,就像你想要拧个螺帽就要把扳手买回来。这样才能使GPU进行深度神经网络的工作,工作速度相较CPU快很多。
参阅该篇博客

2.在配置pytorch虚拟环境中遇到的问题

(1)如何查看电脑能够支持的最高cuda版本

1)鼠标右键,点击NVIDIA控制面板
2)打开左下角系统信息,切换到组件
在这里插入图片描述

(2)在pytorch的虚拟环境中,输入 “torch.cuda.is_available()”,返回true,表示能够使用gpu版的pytorch,但如果在运行程序的过程中出现"GeForce RTX 3090 with CUDA capability sm_86 is not compatible with the current PyTorch installation.",这表示显卡的算力超出了pytorch虚拟环境中cuda支持的算力,此时需要提高cuda的版本

可参阅该篇博客

(3)在配置pytorch虚拟环境前如何选择cuda以及pytorch版本

1)通过(1)可知,在虚拟环境中安装的cuda版本不能大于此版本
2)通过(2)可知,虚拟环境中cuda版本支持的算力应大于等于显卡的算力
不同cuda支持的计算能力情况
不同算力对应显卡列表
3)通过1)、2)确定cuda版本后,根据NVIDIA驱动版本与CUDA版本的对应关系查看是否需要更新自己的显卡驱动
NVIDIA驱动版本与CUDA版本对应关系
4)通过1)、2)确定cuda版本后,根据cuda与pytorch对应的关系确定pytorch版本
在这里插入图片描述
5)根据4)的版本在pytorch官网匹配pytorch的下载命令
PyTorch中torch、torchvision、torchaudio版本对应关系

3.在安装gpu—pytorch中遇到的问题

(1)安装gpu版的pytorch时能够自动安装cuda和cudnn吗

通过命令行会自动将需要的cudatoolkit安装好,不需要单独去在网上搜索怎样安装cuda和cudnn
可参阅该篇博客

(2)电脑会自带cuda吗

一般来说显卡正常工作的话会对应一个cuda版本,系统中的初始cuda也是别人装的
可参阅该篇博客

http://www.lryc.cn/news/11883.html

相关文章:

  • jfr引起的一次jvm异常记录
  • Java智慧校园平台源码:SaaS模式智慧校园运营云平台源码
  • 【yolov5】将标注好的数据集进行划分(附完整可运行python代码)
  • es-05分词器
  • 已解决zipfile.BadZipFile: File is not a zip file
  • Mybatis源码分析:Mybatis的数据存储对象
  • 学习 Python 之 Pygame 开发坦克大战(二)
  • 短视频时代是靠什么赚钱的,介绍常见的5种方式,简单明了
  • 关于CentOS维护的几条简单命令
  • PoW 、PoS , DPoS 算法
  • SpringCloud(PS)远程调用--Feign
  • 2023年全国最新二级建造师精选真题及答案1
  • HydroD 实用教程(四)水动力模型
  • vue项目第七天
  • 拂晓·微信机器人
  • React:Hooks工作机制
  • 基于深度神经网络的3D模型合成【Transformer vs. CNN】
  • 前端面试题整理之HMTL篇(一)
  • 【论文速递】ICLR2018 - 用于小样本语义分割的条件网络
  • 本地生成动漫风格 AI 绘画 图像|Stable Diffusion WebUI 的安装和局域网部署教程
  • 用一行Python代码,为图片上水印版权!
  • java中的lambda表达式
  • 0.1opencv库VS环境配置
  • 第五十七章 树状数组(二)
  • 比特币的网络
  • ChatGPT的模型介绍及GO语言实现API
  • Tile防丢器引入全新防盗模式,苹果Find My功能拓展到大众消费电子
  • 物联网中RocketMQ的使用
  • 用Three.js搭建的一个艺术场景
  • 算法导论【字符串匹配】—朴素算法、Rabin-Karp、有限自动机、KMP