Skip to content

GitLab

Commit 9564de9a authored by dcuai's avatar dcuai
Browse files

Update 24.04.2

parent e496fbfe
Hide whitespace changes
Inline Side-by-side
Showing with 8 additions and 8 deletions
README.md
View file @ 9564de9a
...@@ -21,8 +21,8 @@ CRNN网络将CNN和RNN网络结合，共同训练，使用CNN对输入图像提 ...@@ -21,8 +21,8 @@ CRNN网络将CNN和RNN网络结合，共同训练，使用CNN对输入图像提
### Docker (方法一) ### Docker (方法一)
``` ```
docker pull image.sourcefind.cn:5000/dcu/admin/base/pytorch:1.10.0-centos7.6-dtk-22.10.1-py39-latest docker pull image.sourcefind.cn:5000/dcu/admin/base/vscode-pytorch:2.1.0-ubuntu20.04-dtk24.04.2
docker run -it -v /path/your_code_data/:/path/your_code_data/ --shm-size=32G --privileged=true --device=/dev/kfd --device=/dev/dri/ --group-add video --name docker_name imageID bash docker run -it -v /path/your_code_data/:/path/your_code_data/ -v /opt/hyhal:/opt/hyhal:ro --shm-size=32G --privileged=true --device=/dev/kfd --device=/dev/dri/ --group-add video --name docker_name imageID bash
cd /path/workspace/ cd /path/workspace/
pip3 install -r requirements.txt pip3 install -r requirements.txt
...@@ -33,7 +33,7 @@ pip3 install -r requirements.txt ...@@ -33,7 +33,7 @@ pip3 install -r requirements.txt
``` ```
cd ./docker cd ./docker
docker build --no-cache -t crnn_pytorch:last . docker build --no-cache -t crnn_pytorch:last .
docker run -it -v /path/your_code_data/:/path/your_code_data/ --shm-size=32G --privileged=true --device=/dev/kfd --device=/dev/dri/ --group-add video --name docker_name imageID bash docker run -it -v /path/your_code_data/:/path/your_code_data/ -v /opt/hyhal:/opt/hyhal:ro --shm-size=32G --privileged=true --device=/dev/kfd --device=/dev/dri/ --group-add video --name docker_name imageID bash
``` ```
### Anaconda (方法三) ### Anaconda (方法三)
...@@ -41,10 +41,10 @@ docker run -it -v /path/your_code_data/:/path/your_code_data/ --shm-size=32G --p ...@@ -41,10 +41,10 @@ docker run -it -v /path/your_code_data/:/path/your_code_data/ --shm-size=32G --p
1、关于本项目DCU显卡所需的特殊深度学习库可从光合开发者社区下载安装： https://developer.hpccube.com/tool/ 1、关于本项目DCU显卡所需的特殊深度学习库可从光合开发者社区下载安装： https://developer.hpccube.com/tool/
``` ```
DTK软件栈：dtk22.10 DTK软件栈：dtk24.04.2
python：python3.8 python：3.10
pytorch：1.10.0 pytorch：2.1.0
torchvision:0.10.0 torchvision:0.16.0
``` ```
Tips：以上dtk软件栈、python、pytorch等DCU相关工具版本需要严格一一对应 Tips：以上dtk软件栈、python、pytorch等DCU相关工具版本需要严格一一对应
...@@ -105,7 +105,7 @@ python3 train.py --adadelta --trainRoot ../Datasets/Synth90k/train --valRoot ../ ...@@ -105,7 +105,7 @@ python3 train.py --adadelta --trainRoot ../Datasets/Synth90k/train --valRoot ../
export HSA_FORCE_FINE_GRAIN_PCIE=1 export HSA_FORCE_FINE_GRAIN_PCIE=1
export USE_MIOPEN_BATCHNORM=1 export USE_MIOPEN_BATCHNORM=1
export HIP_VISIBLE_DEVICES=0,1,2,3 export HIP_VISIBLE_DEVICES=0,1,2,3
python -m torch.distributed.launch --nproc_per_node=4 train_ddp.py --adadelta --trainRoot ../Datasets/Synth90k --valRoot ../Datasets/Synth90k --cuda --ngpu 4 --batchSize 64 --workers 8 python -m torch.distributed.launch --nproc_per_node=4 train_ddp.py --adadelta --trainRoot ../Datasets/Synth90k/train --valRoot ../Datasets/Synth90k/val --cuda --ngpu 4 --batchSize 64 --workers 8
``` ```
## 推理 ## 推理
......
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