Update Dockerfile-cuda

mila-iqia · Aug 5, 2024 · 24286c6 · 24286c6
1 parent fdd839c
commit 24286c6
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/docker/Dockerfile-cuda b/docker/Dockerfile-cuda
@@ -52,7 +52,7 @@ RUN apt-get update -y &&\
 # Install Rust
 RUN curl https://sh.rustup.rs -sSf | sh -s -- -y
 ENV PATH="/root/.cargo/bin:${PATH}"
-ENV CUDA_HOME=/usr/local/cuda-11.8
+ENV CUDA_HOME=/usr/local/cuda-12.1
 
 # Install Milabench
 # -----------------