PyTorch环境配置问题

为什么深度学习都是用英伟达的显卡?

首先我们需要了解什么是CUDA?

CUDA(Compute Unified Device Architecture),是显卡厂商 NVIDIA 推出的运算平台。

CUDA就类似于编程语言,开发者和显卡通过CUDA进行交流,开发者用CUDA去操作英伟达显卡,可以给显卡发送各种各样的指令。然而,像PyTorch和Tensorflow这种深度学习的库,本质就是从上一级,调用CUDA的库,随后将我们需要处理的数据交给GPU(显卡)去处理,以此达到加快深度学习网络的训练。

在安装好anaconda之后,开发者需要去根据python的版本、PyTorch的版本、CUDA runtime version版本、CUDA driver version版本。

其中最关键的是,CUDA runtime version 的版本要小于等于 CUDA driver version。

这里CUDA runtime version是支持显卡对应的算力。

通过 anaconda 的 命令行界面输入 nVidia-smi

查找到CUDA runtime version,此处我的是12.4

所以再去pytorch官网找小于12.4的版本。直接下载即可

相关推荐
刀法如飞几秒前
MicroWind:AI编程核心知识库,程序员转型必备
人工智能·aigc·ai编程
2401_865439638 分钟前
CSS如何实现图片自动裁剪填充_巧用object-fit属性控制尺寸
jvm·数据库·python
道可云10 分钟前
AI & OPC 每日资讯(4月15日)|《全球人工智能治理科技社团倡议》发布
人工智能
myan10 分钟前
AI多智能体应用将助推区块链支付崛起
人工智能·区块链
m0_7488394912 分钟前
HTML函数能否用液态金属散热提升性能_极端散热方案实测【汇总】
jvm·数据库·python
2301_8035389513 分钟前
mysql添加索引导致插入变慢怎么办_索引优化与异步处理方案
jvm·数据库·python
该醒醒了~13 分钟前
深度学习异常检测Anomalib算法训练+推理+转化+onnx
人工智能·python·深度学习
OJAC11113 分钟前
站在重构之巅:AI时代的人才成长全路径
人工智能
2301_7826591813 分钟前
如何防止SQL脏数据写入_利用触发器实现强一致性校验
jvm·数据库·python
2301_8176722617 分钟前
如何实现元素从底部进入视口时触发 sticky 定位
jvm·数据库·python