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
MLPerf_ResNet50_tensorflow
Commits
fd36f6ad
Commit
fd36f6ad
authored
Nov 23, 2023
by
liangjing
Browse files
update
parent
9fa9e2d0
Pipeline
#652
failed with stages
in 0 seconds
Changes
3
Pipelines
1
Hide whitespace changes
Inline
Side-by-side
Showing
3 changed files
with
28 additions
and
2 deletions
+28
-2
Dockerfile
Dockerfile
+16
-0
README.md
README.md
+12
-2
result.png
result.png
+0
-0
No files found.
Dockerfile
0 → 100644
View file @
fd36f6ad
# Copyright (c) 2018, NVIDIA CORPORATION. All rights reserved.
#
# Licensed under the Apache License, Version 2.0 (the "License");
# you may not use this file except in compliance with the License.
# You may obtain a copy of the License at
#
# http://www.apache.org/licenses/LICENSE-2.0
#
# Unless required by applicable law or agreed to in writing, software
# distributed under the License is distributed on an "AS IS" BASIS,
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
# See the License for the specific language governing permissions and
# limitations under the License.
FROM
=image.sourcefind.cn:5000/dcu/admin/base/custom:mlperf-maskrcnn-latest
README.md
View file @
fd36f6ad
...
...
@@ -20,6 +20,8 @@ ResNet50是一种用于图像识别的深度神经网络模型,该模型由一
## 环境配置
**Docker (方法一)**
提供
[
光源
](
https://www.sourcefind.cn/#/service-details
)
拉取的训练的docker镜像:
docker pull image.sourcefind.cn:5000/dcu/admin/base/custom:mlperf-resnet50-mpirun-latest
...
...
@@ -28,6 +30,14 @@ ResNet50是一种用于图像识别的深度神经网络模型,该模型由一
# <Container Path>容器映射路径
docker run -it --name mlperf_bert --shm-size=32G --device=/dev/kfd --device=/dev/dri/ --cap-add=SYS_PTRACE --security-opt seccomp=unconfined --ulimit memlock=-1:-1 --ipc=host --network host --group-add video -v <Host Path>:<Container Path> <Image ID> /bin/bash
**Dockerfile (方法二)**
docker build --no-cache -t mlperf_resnet50:latest
docker run -it --name mlperf_resnet50 --shm-size=32G --device=/dev/kfd --device=/dev/dri/ --cap-add=SYS_PTRACE --security-opt seccomp=unconfined --ulimit memlock=-1:-1 --ipc=host --network host --group-add video -v <Host Path>:<Container Path> <Image ID> /bin/bash
# <Image ID>用上面拉取docker镜像的ID替换
# <Host Path>主机端路径
# <Container Path>容器映射路径
镜像版本依赖:
*
DTK驱动:dtk22.10.1
...
...
@@ -45,7 +55,7 @@ ResNet50是一种用于图像识别的深度神经网络模型,该模型由一
具体处理方式可详见:https://github.com/mlcommons/training/tree/master/image_classification
处理好的预训练模型结果如下
:
处理好的预训练模型结果如下
,镜像中已经给出,无需额外下载
mlperf_resnet50
├── checkpoint
...
...
@@ -90,7 +100,7 @@ bash 8dcu_multi.sh >& output.log &
*
https://developer.hpccube.com/codes/modelzoo/mlperf_resnet50_tensorflow
## 参考
## 参考
资料
*
https://mlcommons.org/en/
*
https://github.com/mlcommons
...
...
result.png
View replaced file @
9fa9e2d0
View file @
fd36f6ad
15.9 KB
|
W:
|
H:
566 KB
|
W:
|
H:
2-up
Swipe
Onion skin
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