Change CUDA 12.2 to 12.1 because of missing cuDNN images

This commit is contained in:
Jonatan Kłosko 2023-10-06 12:13:27 +07:00
parent 765d2b6630
commit e2abc760a4
2 changed files with 3 additions and 3 deletions

View file

@ -110,7 +110,7 @@ jobs:
runs-on: ubuntu-latest
strategy:
matrix:
cuda: ["11.8", "12.2"]
cuda: ["11.8", "12.1"]
steps:
- uses: actions/checkout@v3
- run: |

View file

@ -17,10 +17,10 @@ docker buildx build --push --platform linux/amd64,linux/arm64 \
docker/base
docker buildx build --push --platform linux/amd64,linux/arm64 \
-t ghcr.io/livebook-dev/utils:elixir-$elixir-erlang-$otp-cuda12.2 \
-t ghcr.io/livebook-dev/utils:elixir-$elixir-erlang-$otp-cuda12.1 \
--build-arg ELIXIR_VERSION=$elixir \
--build-arg ERLANG_VERSION=$otp \
--build-arg UBUNTU_VERSION=$ubuntu \
--build-arg CUDA_VERSION=12.2.0 \
--build-arg CUDA_VERSION=12.1.0 \
-f docker/base/elixir-cuda.dockerfile \
docker/base