From ab559941bdf46c94e34f0eb83c9c6580e15009d0 Mon Sep 17 00:00:00 2001 From: Ali Mirlou Date: Thu, 24 Nov 2022 17:45:45 +0330 Subject: [PATCH 1/3] Update CUDA version To make it compatible with newer graphic cards --- config.txt | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/config.txt b/config.txt index 226e6f9e..9b8ad9d2 100644 --- a/config.txt +++ b/config.txt @@ -9,4 +9,4 @@ TORCHAUDIO_VERSION=0.11.0 TORCHTEXT_VERSION=0.12.0 TORCHVISION_VERSION=0.12.0 CUDA_MAJOR_VERSION=11 -CUDA_MINOR_VERSION=0 +CUDA_MINOR_VERSION=7 From ae530a5cdac11ffc5abb367084f59da238d388ad Mon Sep 17 00:00:00 2001 From: Ali Mirlou Date: Sat, 26 Nov 2022 02:47:26 +0330 Subject: [PATCH 2/3] Add magma-cuda version --- config.txt | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/config.txt b/config.txt index 9b8ad9d2..ee1de9c9 100644 --- a/config.txt +++ b/config.txt @@ -9,4 +9,5 @@ TORCHAUDIO_VERSION=0.11.0 TORCHTEXT_VERSION=0.12.0 TORCHVISION_VERSION=0.12.0 CUDA_MAJOR_VERSION=11 -CUDA_MINOR_VERSION=7 +CUDA_MINOR_VERSION=4 +MAGMA_CUDA_VERSION=113 From 7a762b3d83e1601f2ea29f1be304ac1d7c399aac Mon Sep 17 00:00:00 2001 From: Ali Mirlou Date: Sat, 26 Nov 2022 02:48:52 +0330 Subject: [PATCH 3/3] Update cudf and cuml --- Dockerfile.tmpl | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/Dockerfile.tmpl b/Dockerfile.tmpl index 6eab7ca0..b9bcf2c8 100644 --- a/Dockerfile.tmpl +++ b/Dockerfile.tmpl @@ -81,7 +81,7 @@ RUN conda config --add channels nvidia && \ # b/232247930: uninstall pyarrow to avoid double installation with the GPU specific version. RUN pip uninstall -y pyarrow && \ - conda install cudf=21.10 cuml=21.10 cudatoolkit=$CUDA_MAJOR_VERSION.$CUDA_MINOR_VERSION && \ + conda install cudf=21.12.02 cuml=21.12 cudatoolkit=$CUDA_MAJOR_VERSION.$CUDA_MINOR_VERSION && \ /tmp/clean-layer.sh {{ end }} @@ -97,7 +97,7 @@ RUN conda install implicit && \ # Install PyTorch {{ if eq .Accelerator "gpu" }} COPY --from=torch_whl /tmp/whl/*.whl /tmp/torch/ -RUN conda install -c pytorch magma-cuda${CUDA_MAJOR_VERSION}${CUDA_MINOR_VERSION} && \ +RUN conda install -c pytorch magma-cuda${MAGMA_CUDA_VERSION} && \ pip install /tmp/torch/*.whl && \ rm -rf /tmp/torch && \ /tmp/clean-layer.sh