Add cuda libraries back

foundation-model-stack · Nov 21, 2023 · 1e2f73f · 1e2f73f
1 parent f16c404
commit 1e2f73f
Showing 1 changed file with 1 addition and 1 deletion.
diff --git 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 && \