调用GPU进行神经网络的训练 GPU环境的搭建

2024-06-07 04:45:33

1. 下载安装CUDA

下载地址：https://developer.nvidia.com/cuda-toolkit-archive
下载界面如图所示。根据自己的Tensorflow版本，选择相应版本的CUDA进行安装，然后根据计算机及操作系统的种类进行进一步的选择 （下载方式建议选择local，network方式可能会出现一些错误）

注意：在进行安装时，选择自定义安装路径，否则会因为临时解压目录和安装目录相同，导致安装结束后，临时解压目录被删除，所以安装目录也被删除。以至于在复制cuDNN文件时找不到文件夹。所以在进行安装时应选择自定义安装

下图所示，为临时解压路径。

下图所示，为安装路径。此处安装路径应自己进行选择，并且不与临时解压路径相同或相包含。

2. 下载安装cuDNN

下载地址：https://developer.nvidia.com/rdp/cudnn-download
下载界面如图所示。首先需要使用NVIDIA账号登录，如果没有需要使用邮箱进行注册。如果忘记密码，可以发送邮件进行重置。
勾选 I Agree To the Terms of cuDNN Software License Agreement，然后选择相应的版本进行下载，Archived cuDNN Releases可以选择其他已有的版本。
下载完成后进行解压，将各个文件夹中的文件分别复制到之前安装cuda的文件夹下（此处注意将同名文件夹中的内容进行复制，如将bin文件夹中的文件复制到cuda的bin文件夹中）：

3.配置环境变量

将CUDA的安装路径设置到Path中

4.Tensorflow的安装

pip指令：pip install tensorflow-gpu  或者在PyCharm中搜索直接进行安装

注意：tensorflow和tensorflow-gpu同时存在可能会存在问题，可以将tensorflow卸载只留下tensorfflow-gpu

调用GPU进行神经网络的训练 GPU环境的搭建相关推荐

利用gpu加速神经网络算法,外接gpu 训练神经网络
神经网络做图像分类一定要用到gpu吗? GPU最大的价值一直是"accelerating"(加速),GPU不是取代CPU,而是利用GPU的并行计算架构,来将并行计算的负载放到GPU ...
训练神经网络gpu占用率低,外接gpu 训练神经网络
为什么要用GPU来训练神经网络而不是CPU? 许多现代神经网络的实现基于GPU,GPU最初是为图形应用而开发的专用硬件组件.所以神经网络收益于游戏产业的发展. 中央处理器(central proces ...
用GPU加速神经网络训练
前言本文记录了做毕设过程中,为了加速神经网络的训练,如何让tf能识别到GPU并加速,加速前resnet架构一个epoch需要跑40分钟,使用mx150(一个laptop的极普通显卡,cuda核心仅有 ...
阿里公开自研AI集群细节：64个GPU，百万分类训练速度提升4倍
点上方蓝字计算机视觉联盟获取更多干货在右上方 ··· 设为星标 ★,与你不见不散编辑:Sophia 计算机视觉联盟报道 | 公众号 CVLianMeng 转载于 :量子位 AI博士笔记系列推 ...
深度神经网络移动终端GPU加速实践
深度神经网络移动终端GPU加速实践前言 AI无疑是近几年的超级风口,"All in AI"等押宝AI的口号层出不穷,AI的风头一时无两.实现AI有很多种途径方法,这其中深度学习神 ...
重磅！Facebook更新PyTorch 1.1，打算跨GPU分割神经网络
时隔半年不到,PyTorch 已经从之前的 1.0 升级到 1.1 版本了.刚刚,Facebook 在年度开发者大会 F8 上宣布正式发布 PyTorch 1.1 版本,这是对 PyTorch 1.0 ...
收藏 | GPU多卡并行训练总结
点击上方"视学算法",选择加"星标"或"置顶" 重磅干货,第一时间送达仅作学术分享,不代表本公众号立场,侵权联系删除转载于:记忆的迷谷@ ...
百度PRNN：增强GPU伸缩性，RNN训练最高提速30倍（源码下载）
尽管有各种深度学习加速器,神经网络的大小依然受限于计算平台的能力.百度硅谷人工智能实验室高级研究员Greg Diamos在最近的ICML 2016上发表了一篇PRNN(Persistent RNNs) ...
重磅！Facebook 更新 PyTorch 1.1，打算跨 GPU 分割神经网络 | 技术头条
作者 | 琥珀出品 | AI科技大本营(ID:rgznai100) 责编 | 胡巍巍时隔半年不到,PyTorch 已经从之前的 1.0 升级到 1.1 版本了.刚刚,Facebook 在年度开发者 ...

最新文章

热门文章