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的版本。直接下载即可

相关推荐
xj7573065334 分钟前
Django 面试常见问题
python·面试·django
jay神4 分钟前
手势识别数据集 - 专业级目标检测训练数据
人工智能·深度学习·yolo·目标检测·计算机视觉
a努力。5 分钟前
得物Java面试被问:Netty的ByteBuf引用计数和内存释放
java·开发语言·分布式·python·面试·职场和发展
海绵宝宝de派小星6 分钟前
AI发展简史与里程碑事件
人工智能·搜索引擎
海绵宝宝de派小星6 分钟前
什么是人工智能?AI、机器学习、深度学习的关系
人工智能·深度学习·机器学习·ai
HaiLang_IT6 分钟前
基于图像处理与注意力机制的输电线路绝缘子缺陷智能识别方法
图像处理·人工智能
大山同学9 分钟前
深度学习任务分类之图像超分辨率
人工智能·深度学习·分类
一招定胜负9 分钟前
机器学习项目:矿物分类系统重制版
人工智能·机器学习·分类
koo3649 分钟前
pytorch深度学习笔记17
pytorch·笔记·深度学习
丝斯201112 分钟前
AI学习笔记整理(53)——大模型之Agent 智能体开发
人工智能·笔记·学习