From 81134a3feaa596e070f0abd3401ff99d12b20252 Mon Sep 17 00:00:00 2001
From: Davide Valsecchi <valsecchi.davide94@gmail.com>
Date: Thu, 16 May 2024 11:46:58 +0200
Subject: [PATCH 1/2] Create Dockerfile_311_cuda

---
 docker/Dockerfile_311_cuda | 38 ++++++++++++++++++++++++++++++++++++++
 1 file changed, 38 insertions(+)
 create mode 100644 docker/Dockerfile_311_cuda

diff --git a/docker/Dockerfile_311_cuda b/docker/Dockerfile_311_cuda
new file mode 100644
index 0000000..8315a38
--- /dev/null
+++ b/docker/Dockerfile_311_cuda
@@ -0,0 +1,38 @@
+FROM python:3.11
+
+# set the workdir
+WORKDIR /root
+
+# minimal software stack
+RUN apt-get update; apt-get clean
+RUN apt-get install -y nano less htop git; apt-get clean
+
+# python software stack
+RUN pip install --no-cache-dir --upgrade pip setuptools
+RUN pip install --no-cache-dir --upgrade ipython
+RUN pip install --no-cache-dir \
+    numpy \
+    scipy \
+    matplotlib \
+    pandas \
+    numexpr \
+    jupyterlab \
+    notebook \
+    scikit-learn \
+    scikit-optimize \
+    tensorflow[and-cuda] \
+    xgboost \
+    scinum \
+    nvidia_smi \
+    py3nvml \  
+    torch
+
+# install cmsml from master
+RUN git clone https://github.com/cms-ml/cmsml.git && \
+    cd cmsml && \
+    pip install --no-cache-dir .[dev] && \
+    cd .. && \
+    rm -rf cmsml
+
+# initial command
+CMD ["bash", "-i", "-l"]

From 8af6e6fd2ddbcca74ac455b15bc6774e43f2bee9 Mon Sep 17 00:00:00 2001
From: Davide Valsecchi <valsecchi.davide94@gmail.com>
Date: Thu, 16 May 2024 11:49:52 +0200
Subject: [PATCH 2/2] Update deploy_images.yml

---
 .github/workflows/deploy_images.yml | 19 +++++++++++++++++++
 1 file changed, 19 insertions(+)

diff --git a/.github/workflows/deploy_images.yml b/.github/workflows/deploy_images.yml
index 9bb323d..b34f3a1 100644
--- a/.github/workflows/deploy_images.yml
+++ b/.github/workflows/deploy_images.yml
@@ -127,7 +127,26 @@ jobs:
           push: ${{ github.event.inputs.skip_push != 'true' }}
           username: ${{ secrets.DOCKERHUB_USERNAME }}
           password: ${{ secrets.DOCKERHUB_TOKEN }}
+          
+  py311-cuda:
+    runs-on: ubuntu-latest
+    steps:
+      - name: Checkout 🛎️
+        uses: actions/checkout@v3
+        with:
+          persist-credentials: false
 
+      - name: Build and deploy image 🐳
+        uses: docker/build-push-action@v1
+        with:
+          path: docker
+          dockerfile: docker/Dockerfile_311_cuda
+          repository: cmsml/cmsml
+          tags: "3.11"
+          push: ${{ github.event.inputs.skip_push != 'true' }}
+          username: ${{ secrets.DOCKERHUB_USERNAME }}
+          password: ${{ secrets.DOCKERHUB_TOKEN }}
+          
   docs:
     needs: py39
     runs-on: ubuntu-latest