Skip to content

Commit 3946c9e

Browse files
authored
Update Dockerfile.base.py38-cuda11.2
1 parent e07b169 commit 3946c9e

File tree

1 file changed

+3
-4
lines changed

1 file changed

+3
-4
lines changed

commons/Dockerfile.base.py38-cuda11.2

Lines changed: 3 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -22,8 +22,7 @@ ENV DEBIAN_FRONTEND=noninteractive \
2222
PIP_IGNORE_INSTALLED=0 \
2323
PYTHONUNBUFFERED=1 \
2424
LIBRARY_PATH=/usr/local/cuda/lib64/stubs \
25-
_CUDA_COMPAT_PATH="/usr/local/cuda/compat" \
26-
LD_LIBRARY_PATH="/usr/local/cuda/extras/CUPTI/lib64:/usr/local/cuda/compat/lib:/usr/local/nvidia/lib:/usr/local/nvidia/lib64:/usr/local/cuda/lib64:/usr/lib/x86_64-linux-gnu:/usr/include/x86_64-linux-gnu:/usr/local/lib" \
25+
_CUDA_COMPAT_PATH="/usr/local/cuda/compat" LD_LIBRARY_PATH="/usr/local/cuda/extras/CUPTI/lib64:/usr/local/cuda/compat/lib:/usr/local/nvidia/lib:/usr/local/nvidia/lib64:/usr/local/cuda/lib64:/usr/lib/x86_64-linux-gnu:/usr/include/x86_64-linux-gnu:/usr/local/lib" \
2726
PATH="/usr/local/nvidia/bin:/usr/local/cuda/bin:/usr/local/mpi/bin:/usr/local/ucx:/usr/local/sbin:/usr/bin/cmake/bin:/usr/local/bin:/usr/sbin:/usr/bin:/sbin:/bin:/usr/local/src/lightgbm/LightGBM" \
2827
CPLUS_INCLUDE_PATH=/usr/include/gdal \
2928
C_INCLUDE_PATH=/usr/include/gdal \
@@ -482,8 +481,8 @@ LABEL ai.backend.kernelspec="1" \
482481
ai.backend.base-distro="ubuntu16.04" \
483482
ai.backend.resource.min.cpu="1" \
484483
ai.backend.resource.min.mem="1g" \
485-
ai.backend.resource.min.cuda.device=1 \
486-
ai.backend.resource.min.cuda.shares=0.1 \
484+
ai.backend.resource.min.cuda.device=0 \
485+
ai.backend.resource.min.cuda.shares=0 \
487486
ai.backend.runtime-type="python" \
488487
ai.backend.runtime-path="/usr/bin/python3" \
489488
ai.backend.service-ports="ipython:pty:3000,jupyter:http:8081,jupyterlab:http:8090,vscode:http:8180,tensorboard:http:6006"

0 commit comments

Comments
 (0)