update

Author: manavmahan

Dockerfile

@@ -1,6 +1,8 @@
 # Use the NVIDIA CUDA base image with CUDA 12.5.1 and cuDNN on Ubuntu 22.04
 FROM nvidia/cuda:12.2.2-cudnn8-runtime-ubuntu22.04
 
+ENV DEBIAN_FRONTEND=noninteractive
+
 # Install Python, pip, and bash
 RUN apt-get update && apt-get install -y \
     python3 \
@@ -9,17 +11,28 @@ RUN apt-get update && apt-get install -y \
 # Set Python3 as the default python
 RUN ln -s /usr/bin/python3 /usr/bin/python
 
+# Set the timezone
+ENV TZ=Europe/Berlin
+
+# Install tzdata and other packages
+RUN apt-get update && \
+    apt-get install -y --no-install-recommends tzdata curl vim && \
+    ln -fs /usr/share/zoneinfo/$TZ /etc/localtime && \
+    dpkg-reconfigure --frontend noninteractive tzdata && \
+    apt-get clean && rm -rf /var/lib/apt/lists/* /tmp/* /var/tmp/*
+
+
 RUN apt-get update && apt-get install -y \
-    bash git-all unzip \
+    bash git-all unzip software-properties-common\
     libgl1-mesa-glx libglib2.0-0 \
     build-essential libopenmpi-dev \
     && rm -rf /var/lib/apt/lists/*
 
 # Install pip, JupyterLab, and attrs
 RUN pip install --upgrade pip
 RUN pip install --upgrade awscli attrs jupyterlab jupyterlab-git
-RUN pip install --upgrade opencv-python flake8 matplotlib pytest shapely tqdm
-RUN pip install --upgrade opencv-python opencv-python torch torch_geometric torchtyping wandb x-transformers
+RUN pip install --upgrade flake8 matplotlib pytest shapely tqdm
+RUN pip install --upgrade opencv-python torch torch_geometric torchtyping wandb x-transformers
 
 # Expose port 8888 for JupyterLab
 EXPOSE 8888

build-tag-push.sh

@@ -1,4 +1,4 @@
 #! /bin/bash
 docker build -t my-gradient-notebook .
-docker tag my-gradient-notebook manavmahan/pytorch-cuda-notebook:latest
-docker push manavmahan/pytorch-cuda-notebook:latest
+docker tag my-gradient-notebook manavmahan/pytorch-cuda-notebook:main
+docker push manavmahan/pytorch-cuda-notebook:main