Skip to content

Change Docker cuda image version from 11.1 to 11.6.2 #838

Merged
merged 6 commits into from
Aug 8, 2022
Merged

Change Docker cuda image version from 11.1 to 11.6.2 #838

Show file tree
Hide file tree
Changes from all commits
Commits
Show all changes
6 commits
Select commit Hold shift + click to select a range
feac709
FIX: cuda version and requierements order
martins0n Aug 4, 2022
6c16c80
FIX: cuda tag
martins0n Aug 4, 2022
a6d8505
FIX: cuda version
martins0n Aug 4, 2022
1dad112
FIX: versions cuda and fail-fast: fasle
martins0n Aug 4, 2022
ed99481
FIX: changelog
martins0n Aug 4, 2022
f161f35
Merge branch 'master' into fix-docker-11.1
martins0n Aug 8, 2022
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
3 changes: 2 additions & 1 deletion .github/workflows/docker-unstable.yml
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -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
Expand Down
3 changes: 2 additions & 1 deletion .github/workflows/publish.yml
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -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
Expand Down
2 changes: 1 addition & 1 deletion CHANGELOG.md
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -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))
-
-

Expand Down
2 changes: 1 addition & 1 deletion docker/README.md
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -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

Expand Down
4 changes: 2 additions & 2 deletions docker/etna-cpu/requirements.txt
View file
Open in desktop
Original file line number Diff line number Diff line change
@@ -1,2 +1,2 @@
etna[all]
jupyter
jupyterlab
etna[all]
4 changes: 2 additions & 2 deletions docker/etna-cuda-10.2/requirements.txt
View file
Open in desktop
Original file line number Diff line number Diff line change
@@ -1,2 +1,2 @@
etna[all]
jupyter
jupyterlab
etna[all]
2 changes: 0 additions & 2 deletions docker/etna-cuda-11.1/requirements.txt
View file
Open in desktop

This file was deleted.

4 changes: 2 additions & 2 deletions docker/etna-cuda-11.1/Dockerfile → docker/etna-cuda-11.6.2/Dockerfile
View file
Open in desktop
Original file line number Diff line number Diff line change
@@ -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}

Expand Down
2 changes: 2 additions & 0 deletions docker/etna-cuda-11.6.2/requirements.txt
View file
Open in desktop
Original file line number Diff line number Diff line change
@@ -0,0 +1,2 @@
jupyterlab
etna[all]