File tree Expand file tree Collapse file tree
phases/00-setup-and-tooling/07-docker-for-ai Expand file tree Collapse file tree Original file line number Diff line number Diff line change @@ -4,23 +4,26 @@ ENV DEBIAN_FRONTEND=noninteractive
44ENV PYTHONUNBUFFERED=1
55
66RUN apt-get update && apt-get install -y --no-install-recommends \
7- python3.12 \
8- python3.12-venv \
9- python3.12-dev \
10- python3-pip \
11- git \
12- curl \
13- build-essential \
7+ software-properties-common \
8+ && add-apt-repository ppa:deadsnakes/ppa \
9+ && apt-get update && apt-get install -y --no-install-recommends \
10+ python3.12 \
11+ python3.12-venv \
12+ python3.12-dev \
13+ git \
14+ curl \
15+ build-essential \
1416 && rm -rf /var/lib/apt/lists/*
1517
1618RUN update-alternatives --install /usr/bin/python python /usr/bin/python3.12 1
1719
18- RUN python -m pip install --no-cache-dir --upgrade pip setuptools wheel
20+ RUN curl -sS https://bootstrap.pypa.io/get-pip.py | python3.12 \
21+ && python -m pip install --no-cache-dir --upgrade pip setuptools wheel
1922
2023RUN python -m pip install --no-cache-dir \
21- torch==2.3 .1 \
22- torchvision==0.18 .1 \
23- torchaudio==2.3 .1 \
24+ torch==2.5 .1 \
25+ torchvision==0.20 .1 \
26+ torchaudio==2.5 .1 \
2427 --index-url https://download.pytorch.org/whl/cu124
2528
2629RUN python -m pip install --no-cache-dir \
@@ -40,4 +43,4 @@ VOLUME ["/workspace", "/models"]
4043
4144EXPOSE 8888
4245
43- CMD ["python" ]
46+ CMD ["python" ]
Original file line number Diff line number Diff line change @@ -157,23 +157,26 @@ ENV DEBIAN_FRONTEND=noninteractive
157157ENV PYTHONUNBUFFERED=1
158158
159159RUN apt-get update && apt-get install -y --no-install-recommends \
160- python3.12 \
161- python3.12-venv \
162- python3.12-dev \
163- python3-pip \
164- git \
165- curl \
166- build-essential \
160+ software-properties-common \
161+ && add-apt-repository ppa:deadsnakes/ppa \
162+ && apt-get update && apt-get install -y --no-install-recommends \
163+ python3.12 \
164+ python3.12-venv \
165+ python3.12-dev \
166+ git \
167+ curl \
168+ build-essential \
167169 && rm -rf /var/lib/apt/lists/*
168170
169171RUN update-alternatives --install /usr/bin/python python /usr/bin/python3.12 1
170172
171- RUN python -m pip install --no-cache-dir --upgrade pip setuptools wheel
173+ RUN curl -sS https://bootstrap.pypa.io/get-pip.py | python3.12 \
174+ && python -m pip install --no-cache-dir --upgrade pip setuptools wheel
172175
173176RUN python -m pip install --no-cache-dir \
174- torch==2.3 .1 \
175- torchvision==0.18 .1 \
176- torchaudio==2.3 .1 \
177+ torch==2.5 .1 \
178+ torchvision==0.20 .1 \
179+ torchaudio==2.5 .1 \
177180 --index-url https://download.pytorch.org/whl/cu124
178181
179182RUN python -m pip install --no-cache-dir \
You can’t perform that action at this time.
0 commit comments