diff --git a/.github/workflows/docker-unstable.yml b/.github/workflows/docker-unstable.yml index efaf97857..cc63b2b52 100644 --- 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 index a8f9b3463..c87d3f14a 100644 --- 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 index 0d85b0463..51f0bcec5 100644 --- 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 index e8a04f569..3837e71ac 100644 --- a/docker/README.md +++ b/docker/README.md @@ -4,7 +4,7 @@ - `ghcr.io/tinkoff-ai/etna/etna-cpu:` - `ghcr.io/tinkoff-ai/etna/etna-cuda-10.2:` -- `ghcr.io/tinkoff-ai/etna/etna-cuda-11.1:` +- `ghcr.io/tinkoff-ai/etna/etna-cuda-11.6.2:` ## Use cases diff --git a/docker/etna-cpu/requirements.txt b/docker/etna-cpu/requirements.txt index b600e26cb..66f675711 100755 --- a/docker/etna-cpu/requirements.txt +++ b/docker/etna-cpu/requirements.txt @@ -1,2 +1,2 @@ -etna[all] -jupyter \ No newline at end of file +jupyterlab +etna[all] \ No newline at end of file diff --git a/docker/etna-cuda-10.2/requirements.txt b/docker/etna-cuda-10.2/requirements.txt index b600e26cb..66f675711 100755 --- a/docker/etna-cuda-10.2/requirements.txt +++ b/docker/etna-cuda-10.2/requirements.txt @@ -1,2 +1,2 @@ -etna[all] -jupyter \ No newline at end of file +jupyterlab +etna[all] \ No newline at end of file diff --git a/docker/etna-cuda-11.1/requirements.txt b/docker/etna-cuda-11.1/requirements.txt deleted file mode 100755 index b600e26cb..000000000 --- a/docker/etna-cuda-11.1/requirements.txt +++ /dev/null @@ -1,2 +0,0 @@ -etna[all] -jupyter \ No newline at end of file diff --git a/docker/etna-cuda-11.1/Dockerfile b/docker/etna-cuda-11.6.2/Dockerfile similarity index 90% rename from docker/etna-cuda-11.1/Dockerfile rename to docker/etna-cuda-11.6.2/Dockerfile index d88ac7e66..0af893ef4 100755 --- a/docker/etna-cuda-11.1/Dockerfile +++ b/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 new file mode 100755 index 000000000..66f675711 --- /dev/null +++ b/docker/etna-cuda-11.6.2/requirements.txt @@ -0,0 +1,2 @@ +jupyterlab +etna[all] \ No newline at end of file