Commit 94c87a82 authored by wanglch's avatar wanglch
Browse files

Update README.md

parent 10f9eff7
...@@ -27,7 +27,7 @@ Donut是一种端到端的(即,自包含的)VDU模型,用于通用理解 ...@@ -27,7 +27,7 @@ Donut是一种端到端的(即,自包含的)VDU模型,用于通用理解
### Docker(方法一) ### Docker(方法一)
[光源](https://www.sourcefind.cn/#/service-details)拉取docker镜像的地址与使用步骤 [光源](https://www.sourcefind.cn/#/service-details)拉取docker镜像的地址与使用步骤
``` ```
docker pull image.sourcefind.cn:5000/dcu/admin/base/pytorch:1.13.1-ubuntu20.04-dtk23.10-py38 docker pull image.sourcefind.cn:5000/dcu/admin/base/pytorch:2.1.0-ubuntu20.04-dtk24.04.1-py3.10
docker run -it -v /path/your_code_data/:/path/your_code_data/ -v /opt/hyhal/:/opt/hyhal/:ro --shm-size=64G --privileged=true --network=host --device=/dev/kfd --device=/dev/dri/ --group-add video --name donut <your imageID> bash docker run -it -v /path/your_code_data/:/path/your_code_data/ -v /opt/hyhal/:/opt/hyhal/:ro --shm-size=64G --privileged=true --network=host --device=/dev/kfd --device=/dev/dri/ --group-add video --name donut <your imageID> bash
...@@ -48,15 +48,14 @@ docker run --shm-size=64G --name donut -v /opt/hyhal:/opt/hyhal:ro --privileged= ...@@ -48,15 +48,14 @@ docker run --shm-size=64G --name donut -v /opt/hyhal:/opt/hyhal:ro --privileged=
关于本项目DCU显卡所需的特殊深度学习库可从[光合](https://developer.hpccube.com/tool/)开发者社区下载安装。 关于本项目DCU显卡所需的特殊深度学习库可从[光合](https://developer.hpccube.com/tool/)开发者社区下载安装。
``` ```
DTK驱动:dtk23.10 DTK驱动:dtk24.04.01
python:python3.8 python:python3.10
torch:1.13.0 torch:2.1.0
torchvision:0.14.0
``` ```
`Tips:以上dtk驱动、python、paddle等DCU相关工具版本需要严格一一对应` `Tips:以上dtk驱动、python、paddle等DCU相关工具版本需要严格一一对应`
``` ```
conda create -n donut python=3.8 conda create -n donut python=3.10
conda activate donut conda activate donut
......
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