saymrwulf/onnxruntime
1
0
Fork 0
mirror of https://github.com/saymrwulf/onnxruntime.git synced 2026-07-30 20:18:08 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

Add share memory Flag in docker (#19672)

Browse source 
### Description



### Motivation and Context
Ref:
https://docs.nvidia.com/deeplearning/frameworks/user-guide/index.html#setincshmem

Co-authored-by: Your Name <your@email.com>
This commit is contained in:
Yi Zhang 2024-02-28 10:40:40 +08:00 committed by GitHub
parent c20ced4132
commit f95c0773a1
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
1 changed files with 1 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

2
tools/ci_build/github/azure-pipelines/linux-gpu-ci-pipeline.yml
View file

@ -241,7 +241,7 @@ stages:
script: |
set -e -x
mkdir -p $HOME/.onnx
docker run --gpus all --rm \
docker run --gpus all --shm-size=1g --ipc=host --ulimit memlock=-1 --ulimit stack=67108864 --rm \
--volume $(Build.SourcesDirectory):/onnxruntime_src \
--volume $(Build.BinariesDirectory)/Release:/build/Release \
--volume /data/models:/build/models:ro \