From 2da490f0611e17c993bf9debaadf5f538b8f6fa9 Mon Sep 17 00:00:00 2001
From: jirka <jirka.borovec@seznam.cz>
Date: Thu, 25 Jul 2024 15:24:54 +0200
Subject: [PATCH] cuda 12.1

---
 .github/workflows/docker-build.yml | 2 +-
 _dockers/ubuntu-cuda/Dockerfile    | 4 ++--
 _requirements/devel.txt            | 3 +++
 3 files changed, 6 insertions(+), 3 deletions(-)

diff --git a/.github/workflows/docker-build.yml b/.github/workflows/docker-build.yml
index 649574c50..0e3783cb7 100644
--- a/.github/workflows/docker-build.yml
+++ b/.github/workflows/docker-build.yml
@@ -17,7 +17,7 @@ concurrency:
   cancel-in-progress: ${{ github.ref != 'refs/heads/main' }}
 
 env:
-  PUSH_DOCKERHUB: ${{ github.ref == 'refs/heads/main' || github.event_name != 'pull_request' }}
+  PUSH_DOCKERHUB: true # ${{ github.ref == 'refs/heads/main' || github.event_name != 'pull_request' }}
 
 jobs:
   build-cuda:
diff --git a/_dockers/ubuntu-cuda/Dockerfile b/_dockers/ubuntu-cuda/Dockerfile
index e00bfb209..95da2df55 100644
--- a/_dockers/ubuntu-cuda/Dockerfile
+++ b/_dockers/ubuntu-cuda/Dockerfile
@@ -13,7 +13,7 @@
 # limitations under the License.
 
 ARG UBUNTU_VERSION=22.04
-ARG CUDA_VERSION=11.8.0
+ARG CUDA_VERSION=12.1.1
 
 
 FROM nvidia/cuda:${CUDA_VERSION}-devel-ubuntu${UBUNTU_VERSION}
@@ -82,7 +82,7 @@ RUN \
     CUDA_VERSION_MM=${CUDA_VERSION%.*} && \
     CU_VERSION_MM=${CUDA_VERSION_MM//'.'/''} && \
     pip install --no-cache-dir -r requirements.txt \
-      --find-links "https://download.pytorch.org/whl/cu${CU_VERSION_MM}/torch_stable.html" && \
+      --extra-index-url="https://download.pytorch.org/whl/cu${CU_VERSION_MM}" && \
     rm -rf requirements.txt .actions/ _requirements/
 
 RUN \
diff --git a/_requirements/devel.txt b/_requirements/devel.txt
index 0dfa5eab3..1ee1b5839 100644
--- a/_requirements/devel.txt
+++ b/_requirements/devel.txt
@@ -11,6 +11,9 @@ papermill>=2.3.4, <2.5.0  # rendering
 matplotlib
 # all is based on PyTorch
 torch
+torchaudio
+torchtext
+torchvision
 
 # fixed version to be able to call it as `python -m kaggle`
 https://github.com/Borda/kaggle-api/archive/refs/heads/setup/python-m.zip