From 1e2f73f100d4a85cb33be40ccfafb52cedc79b0a Mon Sep 17 00:00:00 2001 From: Antoni Viros i Martin Date: Tue, 21 Nov 2023 12:23:55 -0500 Subject: [PATCH] Add cuda libraries back --- nightly-pytorch-compile/Dockerfile | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/nightly-pytorch-compile/Dockerfile b/nightly-pytorch-compile/Dockerfile index 9641d11..10029f5 100644 --- a/nightly-pytorch-compile/Dockerfile +++ b/nightly-pytorch-compile/Dockerfile @@ -12,7 +12,7 @@ RUN apt-get -y -o Acquire::Max-FutureTime=86400 update && \ wget https://developer.download.nvidia.com/compute/cuda/repos/debian11/x86_64/cuda-keyring_1.0-1_all.deb && \ dpkg -i cuda-keyring_1.0-1_all.deb && \ apt-get -y -o Acquire::Max-FutureTime=86400 update && \ - apt-get install -y cuda-compiler-11-8 cuda-nsight-systems-11-8 && \ + apt-get install -y cuda-compiler-11-8 cuda-nsight-systems-11-8 cuda-libraries-11-8 cuda-libraries-dev-11-8 && \ apt-get -y clean RUN chgrp -R 0 /opt/conda && \