深度学习基础环境配置

1.前言

首先简单叙述一下,深度学习环境中安装的软件的作用,首先是Anaconda是一个便捷、高效的环境管理工具,特别是针对Python及其相关包和库的管理。可以自动安装兼容的依赖包。

CUDAToolkit(cuda)是NVIDIA开发的一套软件开发工具包,它为NVIDIA GPU提供了一组API和工具,使得开发人员能够利用GPU的并行计算能力来加速计算密集型应用程序。

cuDNN(CUDA Deep Neural Network library)是NVIDIA开发的一个针对深度神经网络的GPU加速库。它专为深度学习应用设计,提供了高度优化的原语(primitives),用于在NVIDIA GPU上执行深度学习训练和推理中的核心计算。

PyTorch是一个开源的Python机器学习库,特别是作为深度学习框架而广受欢迎。

下面是配置环境比较重要的一点:

先根据自己的项目要求的pytorch版本去确定cuda版本,再确定cudnn版本

安装的流程则是反过来,要先安装cuda——cudnn,最后安装pytorch

列如:open-mmlab的mmpretrain工具箱,根据官方文档给出的安装教程

mmpretr

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值