Skip to content
GitLab
Menu
Projects
Groups
Snippets
Loading...
Help
Help
Support
Community forum
Keyboard shortcuts
?
Submit feedback
Contribute to GitLab
Sign in / Register
Toggle navigation
Menu
Open sidebar
ModelZoo
SSD_pytorch
Commits
da9f8ec5
Commit
da9f8ec5
authored
Nov 05, 2024
by
xinghao
Browse files
Update README.md
parent
e7a9e883
Changes
1
Hide whitespace changes
Inline
Side-by-side
Showing
1 changed file
with
2 additions
and
2 deletions
+2
-2
README.md
README.md
+2
-2
No files found.
README.md
View file @
da9f8ec5
...
...
@@ -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
Write
Preview
Markdown
is supported
0%
Try again
or
attach a new file
.
Attach a file
Cancel
You are about to add
0
people
to the discussion. Proceed with caution.
Finish editing this message first!
Cancel
Please
register
or
sign in
to comment