Commit da9f8ec5 authored by xinghao's avatar xinghao
Browse files

Update README.md

parent e7a9e883
......@@ -16,7 +16,7 @@ SSD通过在每个特征图上直接预测边界框和类别概率,使用多
docker pull image.sourcefind.cn:5000/dcu/admin/base/pytorch:2.1.0-centos7.6-dtk24.04-py310
docker run -it --shm-size 80g --network=host --name=SSD --privileged --device=/dev/kfd --device=/dev/dri --group-add video --cap-add=SYS_PTRACE --security-opt seccomp=unconfined image.sourcefind.cn:5000/dcu/admin/base/pytorch:2.1.0-centos7.6-dtk24.04-py310 bin/bash
```
其它非深度学习库参照requirements.txt安装
安装docker中没有的依赖
```
pip install -r requirements.txt
```
......@@ -78,4 +78,4 @@ python -m torch.distributed.launch --nproc_per_node=4 test.py --config-file conf
https://developer.sourcefind.cn/codes/modelzoo/ssd
## 参考资料
https://github.com/lufficc/SSD
\ No newline at end of file
https://github.com/lufficc/SSD
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