DJL初学者学习笔记(一):Java启动DJL控制台打印No matching cuda flavor for win found: cu65并且在线下载dll文件解决方案

本文介绍了如何使用PyTorch加载和预测深度学习模型,涉及Criteria构建、模型路径设置以及解决CUDA Flavor匹配问题的过程。通过示例代码展示了加载KP Detector和Generator模型的步骤,并提供了遇到CUDA问题时的解决方案,包括检查目录和手动下载所需文件。
摘要由CSDN通过智能技术生成

1.举例,来自知乎深度学习库的事例https://zhuanlan.zhihu.com/p/396524552

2.修改,下载之后做略微的修改指定model文件

        Criteria<NDArray, Map> kpDetector =
                Criteria.builder()
                        .setTypes(NDArray.class, Map.class)
                        .optTranslator(new NPtKTranslator())
                        .optEngine("PyTorch")
                        .optDevice(Device.cpu())
                        .optModelUrls("file:///models/my_model/1/")
                        .optModelName("kp_detector.pt")
                        .build();
        Predictor<NDArray, Map> kPredictor = ModelZoo.loadModel(kpDetector).newPredictor();

       
        //saveBoundingBoxImage(driving_video.get(0));
        Criteria<List, Image> generatorCr =
                Criteria.builder()
                        .setTypes(List.class, Image.class)
                        .optEngine("PyTorch")
                        .optTranslator(new PtGTranslator())
                        .optDevice(Device.cpu())
                        .optModelUrls("file:///models/my_model/1/")
                        .optModelName("generator.pt")
                        //.optModelPath(Paths.get("D:/work/learn/AI_MODEL/generator.pt"))
                        .build();

3.启动之后控制台会打印No matching cuda flavor for win found: cu65并且在线下载dll文件,如果当时网速恰好能在线下载,可忽略此文

4.解决方案:在LibUtils文件中打断点,查看dir的指定路径,没有路径自行新建,查看is地址打开后将符合后续执行逻辑的连接留下来自行下载至dir指定路径

 5.完成4之后重启即可合成视频

 

备注:网速不好难下载的查看这个已下载的

链接:https://pan.baidu.com/s/1yaHo3tB8_CSYQbBu7WTkOA 
提取码:za71 
附:此文仅作为个人学习笔记,希望与各位正在学习的朋友多多交流

  • 4
    点赞
  • 2
    收藏
    觉得还不错? 一键收藏
  • 3
    评论
评论 3
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值