docker: revert nvidia to python base to control version

devitocodes · Aug 3, 2023 · ccba67c · ccba67c
1 parent 67930f5
commit ccba67c
Showing 1 changed file with 1 addition and 4 deletions.
diff --git a/docker/Dockerfile.nvidia b/docker/Dockerfile.nvidia
@@ -8,16 +8,13 @@ ARG arch="nvc"
 ########################################################################
 # Build base image with apt setup and common env
 ########################################################################
-FROM ubuntu:22.04 as sdk-base
+FROM ${pyversion}-slim-bullseye as base
 
 ENV DEBIAN_FRONTEND noninteractive
 
 RUN apt-get update -y && \
     apt-get install -y -q gpg apt-utils curl wget vim libnuma-dev tmux numactl
 
-#Install python
-RUN apt-get update && \
-    apt-get install -y dh-autoreconf python3-venv python3-dev python3-pip    
 
 # nodesource: nvdashboard requires nodejs>=10
 RUN curl https://developer.download.nvidia.com/hpc-sdk/ubuntu/DEB-GPG-KEY-NVIDIA-HPC-SDK | gpg --yes --dearmor -o /usr/share/keyrings/nvidia-hpcsdk-archive-keyring.gpg