# file initially based on https://github.com/huggingface/transformers/tree/master/docker

# FROM nvidia/cuda:11.1.1-cudnn8-devel-ubuntu20.04
# FROM nvidia/cuda:11.8.0-cudnn8-devel-ubuntu22.04
FROM nvidia/cuda:12.6.1-cudnn-devel-ubuntu24.04

RUN apt-get update \
	&& DEBIAN_FRONTEND=noninteractive TZ=Europe/Berlin apt-get install -y tzdata \
	&& cp /usr/share/zoneinfo/Europe/Berlin /etc/localtime \
	&& echo "Europe/Berlin" > /etc/timezone  

RUN apt update && \
	apt install -y bash \
		build-essential \
		git \
		curl \
		ca-certificates \
		python3-dev \
		python3-pip \
		libopenmpi-dev && \
	rm -rf /var/lib/apt/lists

RUN python3 -m pip install --no-cache-dir --break-system-packages \
    torch --index-url https://download.pytorch.org/whl/cu124

RUN git clone https://github.com/NVIDIA/apex
RUN cd apex && \
    pip install -v --disable-pip-version-check --no-cache-dir --no-build-isolation --break-system-packages --config-settings  "--build-option=--cpp_ext" --config-settings "--build-option=--cuda_ext" ./

# fix library path for tensorflow and link a file
ENV LD_LIBRARY_PATH="${LD_LIBRARY_PATH}:/usr/local/cuda/lib64"
RUN ln -s /usr/local/cuda/lib64/libcusolver.so.11 /usr/local/cuda/lib64/libcusolver.so.10

CMD ["/bin/bash"]