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

news/2024/11/28 2:36:39/

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.ppmy.cn/news/26049.html

相关文章

03_Docker 入门

03_Docker 入门 文章目录03_Docker 入门3.1 确保 Docker 已经就绪3.2 运行我们的第一个容器3.3 使用第一个容器3.4 容器命名3.5 重新启动已经停止的容器3.6 附着到容器上3.7 创建守护式容器3.8 容器内部都在干些什么3.9 Docker 日志驱动3.10 查看容器内的进程3.11 Docker 统计信…

千锋教育+计算机四级网络-计算机网络学习-04

UDP概述 UDP协议 面向无连接的用户数据报协议,在传输数据前不需要先建立连接;目地主机的运输层收到UDP报文后,不需要给出任何确认 UDP特点 相比TCP速度稍快些简单的请求/应答应用程序可以使用UDP对于海量数据传输不应该使用UDP广播和多播应用…

【蓝桥日记⑤】2014第五届省赛(软件类)JavaA组❆答案解析

【蓝桥日记⑤】2014第五届省赛(软件类)JavaA组☃答案解析 文章目录【蓝桥日记⑤】2014第五届省赛(软件类)JavaA组☃答案解析1、猜年龄2、李白打酒3、神奇算式4、写日志5、锦标赛6、六角填数7、绳圈8、兰顿蚂蚁9、斐波那契10、波动…

Windows系统扩充C盘空间系列方法总结

目录前言方法一 使用自带的Windows的DiskPart扩充C盘1. 打开cmd2.三步命令方法二:使用Windows系统内置磁盘管理扩展C盘方法三. 使用专业磁盘分区工具总结前言 本教程是总结Windows系统进行C盘(系统盘)扩充空间的系列方法,一般来讲…

用Three.js搭建的一个艺术场景

本文翻译自于Medium,原作者用 Three.js 创建了一个“Synthwave 场景”,效果还不错,在此加上自己的理解,记录一下。在线Demo. 地形构建 作者想要搭建一个中间平坦、两侧有凹凸山脉效果并且能够一直绵延不断的地形,接下…

过滤器和监听器

1、过滤器Filter 作用是防止SQL注入、参数过滤、防止页面攻击、空参数矫正、Token校验、Session验证、点击率统计等等; 使用Filter的步骤 新建类,实现Filter抽象类;重写init、doFilter、destroy方法;在SpringBoot入口中添加注解…

数字经济活动题

讨论活动1-1:数字化 经济数字化(数据数字化,高速通信,大容量存储)将如何影响您所居住的国家?在网上搜索新闻文章(两三篇就够了),并讨论数字化如何影响经济、公共部…

英语基础-状语

1. 课前引语 1. 形容词使用场景 (1). 放在系动词后面作表语 The boy is handsome. (2). 放在名词前面做定语 I like this beautiful girl. (3). 放在宾语后面做补语 You make your father happy. 总结:形容词无论做什么,都离不开名词&#xff0c…