37 Star 223 Fork 52

GVP隐语SecretFlow / secretflow

 / 详情

安装GPU Docker镜像问题后,运行容器时,有4个文件报错和1个模块报错。

待办的
创建于  
2024-02-26 10:44

Issue Type

Build/Install

Source

source

Secretflow Version

1.3.0b0

OS Platform and Distribution

Anolis OS release 8.6

Python version

Python 3.8.17

Bazel version

bazel 1.2.1

GCC/Compiler version

gcc (GCC) 8.5.0 20210514 (Anolis 8.5.0-18.0.4)

What happend and What you expected to happen.

插上英伟达4090GPU加速卡,装上cuda:11.8版本的驱动和GPU Docker镜像,希望能在隐语平台上正确调用GPU加速卡

Reproduction code to reproduce the issue.

GPU Docker镜像后,docker container run,运行容器时,有4个文件报错和1个模块报错。

评论 (6)

xueyanlls 创建了任务

辛苦将报错信息贴出

报错信息看起来是jax的问题,我们有两种获取GPU docker的方法
方式1:从SecretFlow仓库中获取GPU镜像
方式2:自行构建 GPU Docker 镜像#
请问您是通过哪一种呢?

方式1:从SecretFlow仓库中获取GPU镜像

jax的版本太低了,升级后再尝试一下

好的,我们试一下,谢谢

登录 后才可以发表评论

状态
负责人
里程碑
Pull Requests
关联的 Pull Requests 被合并后可能会关闭此 issue
分支
开始日期   -   截止日期
-
置顶选项
优先级
参与者(2)
Python
1
https://gitee.com/secretflow/secretflow.git
git@gitee.com:secretflow/secretflow.git
secretflow
secretflow
secretflow

搜索帮助