深度学习环境搭建【显卡算力、CUDA、pytorch、tensorflow选择】

总结:深度学习环境搭建【显卡算力、CUDA、pytorch、tensorflow选择】

一、了解自己的显卡,确定其对应的算力

不同显卡对应的算力

二、根据算力、显卡驱动,选择cuda版本

显卡驱动与cuda版本对应关系

  • 注意:
    • 如果没有安装显卡驱动,先安装显卡驱动,ubuntu安装显卡驱动
    • 如果已经安装过显卡驱动,根据显卡驱动与cuda版本对应关系,选择cuda。【建议:在显卡驱动支持的范围内,尽量选择高版本的cuda,因为cuda可以向下兼容】

三、pytorch版本选择

进入pytorch官网,选择需要的torch版本安装。

  • 注意:
    • 在显卡支持的范围内,可以自由的选择cuda+torch进行安装,亲测训练目标检测模型可以使用gpu加速
    • 这里讲一下,为什么在显卡驱动支持的范围内,尽量选择高版本的cuda
      • 比如,在定向框检测的情况下,需要创建cuda的扩展(python setup.py develop)。这种情况下就需要考虑电脑上安装的cuda与conda环境中的cudatoolkit或pytorch版本问题
        • 如果电脑上安装的cuda版本高(比如:cuda11.7),conda环境中的cudatoolkit可以低于11.7版本,执行 python setup.py develop 时可以通过;
        • 如果电脑上安装的cuda版本低(比如:cuda10.0),conda环境中的cudatoolkit低于10.0时可以编译通过,高于10.0时则编译失败。

四、tensorflow版本选择

tensorflow与tensorflow-gpu的安装和使用似乎没有pytorch那么挑剔,在conda环境中安装对应的cudatoolkit, cudnn, tensorflow-gpu 即可,安装教程

相关推荐
acstdm1 小时前
DAY 48 CBAM注意力
人工智能·深度学习·机器学习
澪-sl1 小时前
基于CNN的人脸关键点检测
人工智能·深度学习·神经网络·计算机视觉·cnn·视觉检测·卷积神经网络
羊小猪~~2 小时前
数据库学习笔记(十七)--触发器的使用
数据库·人工智能·后端·sql·深度学习·mysql·考研
摸爬滚打李上进2 小时前
重生学AI第十六集:线性层nn.Linear
人工智能·pytorch·python·神经网络·机器学习
HuashuiMu花水木2 小时前
PyTorch笔记1----------Tensor(张量):基本概念、创建、属性、算数运算
人工智能·pytorch·笔记
lishaoan772 小时前
使用tensorflow的线性回归的例子(四)
人工智能·tensorflow·线性回归
lishaoan772 小时前
使用tensorflow的线性回归的例子(九)
tensorflow·线性回归·neo4j
视觉语言导航4 小时前
RAL-2025 | 清华大学数字孪生驱动的机器人视觉导航!VR-Robo:面向视觉机器人导航与运动的现实-模拟-现实框架
人工智能·深度学习·机器人·具身智能
羊小猪~~5 小时前
【NLP入门系列五】中文文本分类案例
人工智能·深度学习·考研·机器学习·自然语言处理·分类·数据挖掘
李师兄说大模型5 小时前
KDD 2025 | 地理定位中的群体智能:一个多智能体大型视觉语言模型协同框架
人工智能·深度学习·机器学习·语言模型·自然语言处理·大模型·deepseek