Skip to content

Commit

Permalink
Update Dockerfile_ubuntu_2204_tf_cuda_11 (#544)
Browse files Browse the repository at this point in the history
  • Loading branch information
baek-jinoo authored Aug 1, 2023
1 parent 84b2adb commit c8e97df
Showing 1 changed file with 10 additions and 8 deletions.
18 changes: 10 additions & 8 deletions perfzero/docker/Dockerfile_ubuntu_2204_tf_cuda_11
Original file line number Diff line number Diff line change
Expand Up @@ -6,7 +6,7 @@ FROM nvidia/cuda:11.8.0-cudnn8-devel-ubuntu22.04 as base
ARG tensorflow_pip_spec="tf-nightly"
ARG local_tensorflow_pip_spec=""
ARG extra_pip_specs=""
ENV PIP_CMD="python3.9 -m pip"
ENV PIP_CMD="python3.11 -m pip"

# setup.py passes the base path of local .whl file is chosen for the docker image.
# Otherwise passes an empty existing file from the context.
Expand Down Expand Up @@ -40,20 +40,22 @@ RUN apt-get update && apt-get install -y --no-install-recommends \
lsb-core \
curl

# Python 3.9 related deps in this ppa.
# Python 3.11 related deps in this ppa.

RUN add-apt-repository ppa:deadsnakes/ppa


# Install / update Python and Python3
RUN apt-get install -y --no-install-recommends \
python3.9 \
python3.11 \
python3-pip \
python3.9-dev \
python3.11-dev \
python3-setuptools \
python3.9-venv \
python3.9-distutils \
python3.9-lib2to3

python3.11-venv \
python3.11-distutils \
python3.11-lib2to3


# Upgrade pip, need to use pip3 and then pip after this or an error
# is thrown for no main found.
RUN ${PIP_CMD} install --upgrade pip
Expand Down

0 comments on commit c8e97df

Please sign in to comment.