Skip to content

Commit d246796

Browse files
committed
Add TF 2.15 container setup and update requirements
1 parent a1f4171 commit d246796

File tree

2 files changed

+48
-0
lines changed

2 files changed

+48
-0
lines changed

Dockerfile

Lines changed: 36 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,36 @@
1+
# Local Usage:
2+
# ```
3+
# docker build -t ghcr.io/bouncmpe/cuda-python3 containers/cuda-python3/
4+
# docker run -it --rm --gpus=all ghcr.io/bouncmpe/cuda-python3
5+
# ```
6+
7+
# Base image
8+
FROM nvidia/cuda:12.2.2-runtime-ubuntu22.04
9+
10+
LABEL maintainer="Kutay Eroglu"
11+
12+
ENV LANG=C.UTF-8 LC_ALL=C.UTF-8
13+
14+
RUN apt-get update && \
15+
export DEBIAN_FRONTEND=noninteractive && \
16+
apt-get -y install --no-install-recommends \
17+
git \
18+
wget \
19+
cmake \
20+
ninja-build \
21+
build-essential \
22+
python3 \
23+
python3-dev \
24+
python3-pip \
25+
python-is-python3 && \
26+
# Upgrade pip
27+
python3 -m pip install --no-cache-dir --upgrade pip && \
28+
# Clean up APT caches to reduce image size
29+
apt-get autoremove -y && \
30+
apt-get clean -y && \
31+
rm -rf /var/lib/apt/lists/*
32+
33+
34+
COPY requirements.txt .
35+
36+
RUN python3 -m pip install --no-cache-dir -r requirements.txt

requirements.txt

Lines changed: 12 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,12 @@
1+
tensorflow==2.15.0
2+
numpy>=1.23
3+
pandas>=1.4
4+
scipy>=1.7
5+
pillow>=9.0
6+
matplotlib>=3.5
7+
seaborn>=0.12
8+
imageio>=2.30
9+
jupyter>=1.0
10+
neptune==1.10.4
11+
# tensorflow-addons is not guaranteed to support TF 2.15; include only if needed
12+

0 commit comments

Comments
 (0)