Commit b390e927 authored by dcuai's avatar dcuai
Browse files

更换dtk24.04.1镜像

parent 9c83ae9f
...@@ -36,15 +36,15 @@ pip install -r requirement.txt -i https://mirrors.aliyun.com/pypi/simple/ --tru ...@@ -36,15 +36,15 @@ pip install -r requirement.txt -i https://mirrors.aliyun.com/pypi/simple/ --tru
关于本项目DCU显卡所需的特殊深度学习库可从[光合](https://developer.hpccube.com/tool/)开发者社区下载安装。 关于本项目DCU显卡所需的特殊深度学习库可从[光合](https://developer.hpccube.com/tool/)开发者社区下载安装。
``` ```
DTK驱动:dtk23.10 DTK驱动:dtk24.04.1
python:python3.8 python:python3.10
torch:2.1.0 torch:2.1.0
torchvision:0.16.0 torchvision:0.16.0
triton:2.1.0 triton:2.1.0
apex:0.1 apex:1.1.0
deepspeed:0.12.3 deepspeed:0.12.3
flash_attn:2.0.4 flash_attn:2.0.4
xformers:0.0.23 xformers:0.0.25
``` ```
`Tips:以上dtk驱动、python、torch等DCU相关工具版本需要严格一一对应` `Tips:以上dtk驱动、python、torch等DCU相关工具版本需要严格一一对应`
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment