MeteoInfo的个人博客分享 http://blog.sciencenet.cn/u/MeteoInfo

博文

在DL4J中使用GPU

已有 2614 次阅读 2020-12-29 22:45 |个人分类:机器学习|系统分类:科研笔记

DL4J是Java语言开发的深度学习库,底层的ND4J支持CPU和GPU,今天尝试了GPU设置:

  <dependency>
       <groupId>org.nd4j</groupId>
       <artifactId>nd4j-cuda-10.2-platform</artifactId>
       <version>1.0.0-beta7</version>
   </dependency>

在Window 10 PC机上安装了CUDA 10.2,然后尝试了mnist数据集的训练程序,总是弹出 jnind4jcuda.dll: Can't find dependent libraries 之类的错误信息,无法运行。多番查找之后搜到了这个帖子:https://community.konduit.ai/t/error-after-upgrade-to-1-0-0-beta7/676  。发现在System32目录中缺少了 vcruntime140_1.dll 这个文件。下载该文件并拷贝入System32目录后运行程序通过。



https://blog.sciencenet.cn/blog-611158-1264800.html

上一篇:发布MeteoInfo 2.4.0
下一篇:发布MeteoInfo 3.2.0
收藏 IP: 123.119.112.*| 热度|

0

该博文允许注册用户评论 请点击登录 评论 (0 个评论)

数据加载中...

Archiver|手机版|科学网 ( 京ICP备07017567号-12 )

GMT+8, 2024-9-27 07:38

Powered by ScienceNet.cn

Copyright © 2007- 中国科学报社

返回顶部