cuda 12.1

.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:

_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 \

_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