本文介绍了如何使用CUDA 9.1和CuDNN 7.1来加速Deeplearning4j(DL4J)模型的训练。首先，详细说明了卸载旧版CUDA和安装CUDA 9.1的步骤，以及在安装过程中应注意的事项。然后，展示了如何安装CuDNN 7.1并将其与DL4J整合以实现加速。通过修改POM文件添加相关依赖，可以成功运行GPU加速的DL4J程序，即便在遇到某些报错的情况下，程序仍能正常运行。最后，提供了Lenet模型用于手写数字识别的示例，并提及了更高的硬件配置将带来更显著的加速效果。
Deeplearning4j-使用Cuda 9.1和 Cudnn7.1 加速模型训练
一、卸载Cuda （可选）
我本机原本安装的版本为 Cuda 8.0，因为Dl4j更新版本之后，支持 Cuda 9.1，因此需要先对原有软件进行卸载。
我电脑的操作系统为win 10，在你安装完成以后，会有如下图所示的安装软件：
除了图中用红框标注的这三个，全部卸载。
即可完成cuda的卸载。
二、安装Cuda。
下载地址：
链接：https://pan.baidu.com/s/14yvW1C3M32TZyeN-kRXEyw 密码：z9k6
为了保证结果的可复现。 Cuda和Cudnn的安装地址已经放在上面了。
在安装的时候，需要注意使用自定义安装，
在安装的时候需要勾掉以下属性
因为你不是 cuda 开发人员，只是使用的用户，因此不需要以下三项：
- Documentation: cuda开发文档
- Samples: cuda示例
- VS Studio Integration: VS开发cuda的集成插件。
也不安装Driver components，是害怕和你现有的软件冲突，导致显示器显示不正常。
在无限下一步之后安装完毕，在 CMD 窗口中使用nvcc -V命令查看 cuda 版本。
三、使用Cuda9.1加速 dl4j
dl4j使用gpu后端加速非常容易，只需要切换pom文件为：
<dependency>
<groupId>org.nd4j</
最低0.47元/天 解锁文章
1146
被折叠的 条评论
为什么被折叠?
到【灌水乐园】发言
