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
关于本项目DCU显卡所需的特殊深度学习库可从[光合](https://developer.hpccube.com/tool/)开发者社区下载安装。
```
DTK驱动:dtk23.10
python:python3.8
DTK驱动:dtk24.04.1
python:python3.10
torch:2.1.0
torchvision:0.16.0
triton:2.1.0
apex:0.1
apex:1.1.0
deepspeed:0.12.3
flash_attn:2.0.4
xformers:0.0.23
xformers:0.0.25
```
`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