Enable CUDA_VISIBLE_DEVICES and so on

mlcommons · Feb 2, 2023 · 96a475f · 96a475f
1 parent 0a085c1
commit 96a475f
Show file tree

Hide file tree

Showing 4 changed files with 7 additions and 7 deletions.
diff --git a/Dockerfile-CUDA11.3 b/Dockerfile-CUDA11.3
@@ -15,7 +15,7 @@ RUN python3.8 -m pip install openvino-dev==2022.1.0 opencv-python-headless mlcub
 # Do some dependency installation separately here to make layer caching more efficient
 COPY ./setup.py ./setup.py
 RUN python3.8 -c "from setup import requirements; file = open('requirements.txt', 'w'); file.writelines([req + '\n' for req in requirements]); file.close()" \
-    && python 3.8 -m pip install -r ./requirements.txt
+    && python3.8 -m pip install -r ./requirements.txt
 
 COPY . /GaNDLF
 WORKDIR /GaNDLF

diff --git a/Dockerfile-ROCm b/Dockerfile-ROCm
@@ -15,7 +15,7 @@ RUN apt-get update && apt-get install -y libgl1
 # Do some dependency installation separately here to make layer caching more efficient
 COPY ./setup.py ./setup.py
 RUN python3.8 -c "from setup import requirements; file = open('requirements.txt', 'w'); file.writelines([req + '\n' for req in requirements]); file.close()" \
-    && python 3.8 -m pip install -r ./requirements.txt
+    && python3.8 -m pip install -r ./requirements.txt
 
 COPY . /GaNDLF
 WORKDIR /GaNDLF

diff --git a/mlcube/mlcube.yaml b/mlcube/mlcube.yaml
@@ -34,8 +34,8 @@ docker:
 
   # These settings should be set by global MLCube configuration, generally not per-deployment.
   # However, some sane defaults (for Docker >19.03) are here:
-  # env_args: CUDA_VISIBLE_DEVICES=$CUDA_VISIBLE_DEVICES
-  # gpu_args: --gpus all
+  env_args: CUDA_VISIBLE_DEVICES=$CUDA_VISIBLE_DEVICES
+  gpu_args: --gpus all
 
 
 singularity:

diff --git a/mlcube/mlcube_medperf.yaml b/mlcube/mlcube_medperf.yaml
@@ -32,10 +32,10 @@ docker:
   # Docker file name within docker build context. Any "Dockerfile-*" in the GaNDLF source repo is valid.
   build_file: "Dockerfile-CUDA11.3"
 
-  # These settings should be set by global MLCube configuration, generally not per-deployment.
+  # These settings should usually be set by global MLCube configuration, generally not per-deployment.
   # However, some sane defaults (for Docker >19.03) are here:
-  # env_args: CUDA_VISIBLE_DEVICES=$CUDA_VISIBLE_DEVICES
-  # gpu_args: --gpus all
+  env_args: CUDA_VISIBLE_DEVICES=$CUDA_VISIBLE_DEVICES
+  gpu_args: --gpus all
 
 
 singularity: