Commit f7e394b6 authored by panhb's avatar panhb
Browse files

Update README.md

parent 5b036548
Pipeline #1679 canceled with stages
......@@ -76,8 +76,8 @@ squad-v1.1 eval脚本:[evaluate-v1.1.py](https://github.com/allenai/bi-att-flo
### Docker(方式一)
```
docker pull image.sourcefind.cn:5000/dcu/admin/base/tensorflow:2.7.0-centos7.6-dtk-22.10.1-py37-latest
docker run -dit --network=host --name=bert_tensorflow --privileged --device=/dev/kfd --device=/dev/dri --ipc=host --shm-size=16G --group-add video --cap-add=SYS_PTRACE --security-opt seccomp=unconfined -u root --ulimit stack=-1:-1 --ulimit memlock=-1:-1 image.sourcefind.cn:5000/dcu/admin/base/tensorflow:2.7.0-centos7.6-dtk-22.10.1-py37-latest
docker pull image.sourcefind.cn:5000/dcu/admin/base/tensorflow:2.13.1-ubuntu20.04-dtk24.04.1-py3.10
docker run -dit --network=host --name=bert_tensorflow --privileged --device=/dev/kfd --device=/dev/dri --ipc=host --shm-size=16G --group-add video --cap-add=SYS_PTRACE --security-opt seccomp=unconfined -u root --ulimit stack=-1:-1 --ulimit memlock=-1:-1 image.sourcefind.cn:5000/dcu/admin/base/tensorflow:2.13.1-ubuntu20.04-dtk24.04.1-py3.10
docker exec -it bert_tensorflow /bin/bash
pip install -r requirements.txt
```
......@@ -94,7 +94,7 @@ pip install -r requirements.txt
### Conda(方式三)
```
conda create -n bert_tensorflow python=3.7
conda create -n bert_tensorflow python=3.10
pip install -r requirements.txt
```
......@@ -182,6 +182,10 @@ sh bert_class.sh
--model_dir 模型保存文件夹
--distribution_strategy 分布式策略
--num_gpus 使用gpu数量
修改库版本:
absl==1.4.0
tf-models-official==2.6.0
```
### 多卡运行-MNLI
......
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