Change Docker cuda image version from 11.1 to 11.6.2 (#838)

tinkoff-ai · Aug 8, 2022 · 3c036d2 · 3c036d2 · github-actions · Aug 8, 2022
1 parent 71a19db
commit 3c036d2
Show file tree

Hide file tree

Showing 9 changed files with 14 additions and 12 deletions.
diff --git a/.github/workflows/docker-unstable.yml b/.github/workflows/docker-unstable.yml
@@ -12,11 +12,12 @@ jobs:
     runs-on: ubuntu-latest
 
     strategy:
+      fail-fast: false
       matrix:
         dockerfile:
           - {"name": etna-cpu, "path": docker/etna-cpu/Dockerfile}
           - {"name": etna-cuda-10.2, "path": docker/etna-cuda-10.2/Dockerfile}
-          - {"name": etna-cuda-11.1, "path": docker/etna-cuda-11.1/Dockerfile}
+          - {"name": etna-cuda-11.6.2, "path": docker/etna-cuda-11.6.2/Dockerfile}
 
     steps:
       - uses: actions/checkout@v2

diff --git a/.github/workflows/publish.yml b/.github/workflows/publish.yml
@@ -93,11 +93,12 @@ jobs:
     runs-on: ubuntu-latest
 
     strategy:
+      fail-fast: false
       matrix:
         dockerfile:
           - {"name": etna-cpu, "path": docker/etna-cpu/Dockerfile}
           - {"name": etna-cuda-10.2, "path": docker/etna-cuda-10.2/Dockerfile}
-          - {"name": etna-cuda-11.1, "path": docker/etna-cuda-11.1/Dockerfile}
+          - {"name": etna-cuda-11.6.2, "path": docker/etna-cuda-11.6.2/Dockerfile}
 
     steps:
       - uses: actions/checkout@v2

diff --git a/CHANGELOG.md b/CHANGELOG.md
@@ -43,7 +43,7 @@ and this project adheres to [Semantic Versioning](https://semver.org/spec/v2.0.0
 - 
 - 
 - 
-- 
+- Change Docker cuda image version from 11.1 to 11.6.2 ([#838](https://github.com/tinkoff-ai/etna/pull/838))
 - 
 - 
 

diff --git a/docker/README.md b/docker/README.md
@@ -4,7 +4,7 @@
 
 - `ghcr.io/tinkoff-ai/etna/etna-cpu:<tag>`
 - `ghcr.io/tinkoff-ai/etna/etna-cuda-10.2:<tag>`
-- `ghcr.io/tinkoff-ai/etna/etna-cuda-11.1:<tag>`
+- `ghcr.io/tinkoff-ai/etna/etna-cuda-11.6.2:<tag>`
 
 ## Use cases
 

diff --git a/docker/etna-cpu/requirements.txt b/docker/etna-cpu/requirements.txt
@@ -1,2 +1,2 @@
-etna[all]
-jupyter
+jupyterlab
+etna[all]
diff --git a/docker/etna-cuda-10.2/requirements.txt b/docker/etna-cuda-10.2/requirements.txt
@@ -1,2 +1,2 @@
-etna[all]
-jupyter
+jupyterlab
+etna[all]
diff --git a/docker/etna-cuda-11.1/requirements.txt b/docker/etna-cuda-11.1/requirements.txt
diff --git a/docker/etna-cuda-11.1/Dockerfile → docker/etna-cuda-11.6.2/Dockerfile b/docker/etna-cuda-11.1/Dockerfile → docker/etna-cuda-11.6.2/Dockerfile
@@ -1,6 +1,6 @@
-ARG BASE_IMAGE=nvidia/cuda:11.1-cudnn8-runtime-ubuntu18.04
+ARG BASE_IMAGE=nvidia/cuda:11.6.2-cudnn8-runtime-ubuntu18.04
 ARG DISTRO_ARCH=ubuntu1804/x86_64
-ARG CUDA_VERSION_FOR_TORCH=cu111
+ARG CUDA_VERSION_FOR_TORCH=cu116
 
 FROM ${BASE_IMAGE}
 

diff --git a/docker/etna-cuda-11.6.2/requirements.txt b/docker/etna-cuda-11.6.2/requirements.txt
@@ -0,0 +1,2 @@
+jupyterlab
+etna[all]