1
5
框架组件
docker运行deepspeed报错
2024-10-24 10:45:12
107

docker image: docker pull image.sourcefind.cn:5000/dcu/admin/base/jupyterlab-deepspeed:0.9.2-centos7.6-dtk23.04-py38-scnet

rocBLAS error: Tensile solution found, but exception thrown for { a_type: "f16_r", b_type: "f16_r", c_type: "f16_r", d_type: "f16_r", compute_type: "f16_r", transA: 'T', transB: 'N', M: 1024, N: 1024, K: 1024, alpha: 1, row_stride_a: 1, col_stride_a: 1024, row_stride_b: 1, col_stride_b: 1024, row_stride_c: 1, col_stride_c: 1024, row_stride_d: 1, col_stride_d: 1024, beta: 0, batch_count: 1, strided_batch: true, stride_a: 1, stride_b: 1, stride_c: 1, stride_d: 1, atomics_mode: atomics_allowed }
Kernel Cijk_Alik_Bljk_HB_MT128x128x32_SE_APM1_AF0EM2_AF1EM1_AMAS3_ASAE01_ASCE01_ASEM2_BL1_DTL0_ETSP_EPS1_FL0_GRVW8_GSU1_GSUAMB_ISA906_IU1_K1_KLA_LPA0_LPB0_LDL1_LRVW8_MAC_MDA2_NLCA1_NLCB1_ONLL1_PK0_PGR1_PLR1_RK0_SU32_SUM0_SUS256_SVW4_SNLL0_TT8_8_USFGRO0_VAW2_VS1_VW8_WG16_16_1_WGM8 not found in any loaded module.

测试脚本见附件

全部评论(5)
只看楼主
R
renzhc
1楼
可以试试换用较高版本的dtk或较新的镜像
2024/10/24
R
renzhc
回复 @renzhc
这边测试在镜像内换用23.10的dtk就可以解决,也可以使用更新的版本。
2024/10/24
网
网友KcWCRt
回复 @renzhc
请问如何升级dtk,有没有文档呢?
2024/10/25
R
renzhc
回复 @renzhc
可以在资源下载中心下载您需要的dtk,解压后执行source {DTK解压目录}/env.sh即可;也可以替换掉/opt目录下的dtk软连接。更详细的您可以参照https://developer.sourcefind.cn/gitbook//dcu_developer/OperationManual/2_InstallROCm/InstallROCm.html
2024/10/25
全部4条回复
没有更多内容了
--
楼主
Level.3
0 帖子
楼主热帖
暂无数据