关于构建torch_npu2.3.1docker的过程
收藏回复举报
关于构建torch_npu2.3.1docker的过程
t('forum.solved') 已解决
发表于2024-07-29 13:31:42
0 查看

构建基于cann8 与torch2.3.1(torch_npu 2.3.1)的docker时,需要注意一下问题:

1. 相比于其他版本,需要安装libsqlite3-dev包,之前的版本好像没有遇到这个问题。

2. numpy的版本需要小于2,我指定的版本的是1.26.2,可以运行。

我们的npu机器不能联网,需要通过部署docker的方式进行运行,每次需要把构建好的docker容器保存出来,再传到npu的机器上使用,以上两个问题浪费了我一天的时间,如果有人遇到同样的问题,希望可以减少你们部署环境的时间。

PS:ascend官方能不能出一个自己构建docker的指南,或者手册。

本帖最后由 匿名用户2024/07/30 08:09:49 编辑

我要发帖子