Try removing --privileged flag

hcho3 · Nov 17, 2024 · 49eaec1 · 49eaec1
1 parent 11b0427
commit 49eaec1
Show file tree

Hide file tree

Showing 3 changed files with 3 additions and 6 deletions.
diff --git a/ops/pipeline/test-cpp-gpu.sh b/ops/pipeline/test-cpp-gpu.sh
@@ -17,7 +17,6 @@ case "${arg}" in
     python3 ops/docker_run.py --container-id xgb-ci.gpu --use-gpus \
       -- nvidia-smi
     python3 ops/docker_run.py --container-id xgb-ci.gpu --use-gpus \
-      --run-args='--privileged' \
       -- build/testxgboost
     ;;
 
@@ -26,7 +25,6 @@ case "${arg}" in
     python3 ops/docker_run.py --container-id xgb-ci.gpu --use-gpus \
       -- nvidia-smi
     python3 ops/docker_run.py --container-id xgb-ci.gpu --use-gpus \
-      --run-args='--privileged' \
       -- build/testxgboost --use-rmm-pool
     ;;
 
@@ -35,7 +33,7 @@ case "${arg}" in
     python3 ops/docker_run.py --container-id xgb-ci.gpu --use-gpus \
       -- nvidia-smi
     python3 ops/docker_run.py --container-id xgb-ci.gpu --use-gpus \
-      --run-args='--shm-size=4g --privileged' \
+      --run-args='--shm-size=4g' \
       -- build/testxgboost --gtest_filter=*MGPU*
     ;;
 

diff --git a/ops/pipeline/test-jvm-gpu.sh b/ops/pipeline/test-jvm-gpu.sh
@@ -14,5 +14,5 @@ set -x
 python3 ops/docker_run.py --container-id xgb-ci.jvm_gpu_build --use-gpus \
   -- nvidia-smi
 python3 ops/docker_run.py --container-id xgb-ci.jvm_gpu_build --use-gpus \
-  --run-args "-e SCALA_VERSION=${SCALA_VERSION} -e USE_CUDA=1 -e SKIP_NATIVE_BUILD=1 --privileged" \
+  --run-args "-e SCALA_VERSION=${SCALA_VERSION} -e USE_CUDA=1 -e SKIP_NATIVE_BUILD=1" \
   -- ops/pipeline/build-test-jvm-packages-impl.sh
diff --git a/ops/pipeline/test-python.sh b/ops/pipeline/test-python.sh
@@ -38,7 +38,6 @@ case "$suite" in
     python3 ops/docker_run.py --container-id "${container_id}" --use-gpus \
       -- nvidia-smi
     python3 ops/docker_run.py --container-id "${container_id}" --use-gpus \
-      --run-args='--privileged' \
       -- bash test-python-wrapper.sh gpu_test
     ;;
 
@@ -56,7 +55,7 @@ case "$suite" in
     python3 ops/docker_run.py --container-id "${container_id}" --use-gpus \
       -- nvidia-smi
     python3 ops/docker_run.py --container-id "${container_id}" --use-gpus \
-      --run-args='--shm-size=4g --privileged' \
+      --run-args='--shm-size=4g' \
       -- bash test-python-wrapper.sh gpu_test
     ;;