2025-03-13 15:21:06 +08:00
|
|
|
FROM tensorflow/tensorflow:latest-gpu
|
|
|
|
|
|
|
|
|
|
# Install zstd, wget, python3-pip, and ngrok dependencies
|
|
|
|
|
RUN apt-get update && apt-get install -y zstd wget python3-pip unzip
|
|
|
|
|
|
|
|
|
|
# Download and set up CUDA repository pin
|
|
|
|
|
RUN wget https://developer.download.nvidia.com/compute/cuda/repos/ubuntu2204/x86_64/cuda-ubuntu2204.pin && \
|
|
|
|
|
mv cuda-ubuntu2204.pin /etc/apt/preferences.d/cuda-repository-pin-600
|
|
|
|
|
|
|
|
|
|
# Download and install CUDA repository
|
|
|
|
|
RUN wget https://developer.download.nvidia.com/compute/cuda/12.3.2/local_installers/cuda-repo-ubuntu2204-12-3-local_12.3.2-545.23.08-1_amd64.deb && \
|
|
|
|
|
dpkg -i cuda-repo-ubuntu2204-12-3-local_12.3.2-545.23.08-1_amd64.deb && \
|
|
|
|
|
cp /var/cuda-repo-ubuntu2204-12-3-local/cuda-*-keyring.gpg /usr/share/keyrings/ && \
|
|
|
|
|
apt-get update && \
|
|
|
|
|
apt-get -y install cuda-toolkit-12-3
|
|
|
|
|
|
|
|
|
|
# Download and install cuDNN repository
|
|
|
|
|
RUN wget https://developer.download.nvidia.com/compute/cudnn/9.7.1/local_installers/cudnn-local-repo-ubuntu2204-9.7.1_1.0-1_amd64.deb && \
|
|
|
|
|
dpkg -i cudnn-local-repo-ubuntu2204-9.7.1_1.0-1_amd64.deb && \
|
|
|
|
|
cp /var/cudnn-local-repo-ubuntu2204-9.7.1/cudnn-*-keyring.gpg /usr/share/keyrings/ && \
|
|
|
|
|
apt-get update && \
|
|
|
|
|
apt-get -y install cudnn
|
|
|
|
|
|
|
|
|
|
# Set environment variables
|
|
|
|
|
ENV CUDA_HOME=/usr/local/cuda
|
|
|
|
|
ENV LD_LIBRARY_PATH=${CUDA_HOME}/lib64:${LD_LIBRARY_PATH}
|
|
|
|
|
ENV PATH=${CUDA_HOME}/bin:${PATH}
|
|
|
|
|
|
|
|
|
|
# Verify the installation
|
2025-03-21 17:41:49 +08:00
|
|
|
# RUN nvcc --version
|
|
|
|
|
# RUN nvidia-smi
|
2025-03-13 15:21:06 +08:00
|
|
|
|
|
|
|
|
# Install Python packages
|
|
|
|
|
RUN pip install sqlalchemy scikit-learn pandas matplotlib kaggle
|
