Merge pull request #19 from foundation-model-stack/test-gpl

Test gpl
foundation-model-stack · Oct 25, 2023 · d3d74f5 · d3d74f5
2 parents 7d120f0 + f16c404
commit d3d74f5
Showing 1 changed file with 5 additions and 3 deletions.
diff --git a/nightly-pytorch-compile-clang/Dockerfile b/nightly-pytorch-compile-clang/Dockerfile
@@ -6,14 +6,16 @@ ENV HOME=/homedir \
 
 # tools
 RUN apt-get -y -o Acquire::Max-FutureTime=86400 update && \
+    apt-get install -y gnupg vim nmon software-properties-common && \
     apt-key del 7fa2af80 && \
     add-apt-repository contrib && \
     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 gnupg vim nmon clang software-properties-common && \
-    apt-get install -y cuda-compiler-12-1 cuda-nsight-systems-12-1 && \
-    apt-get -y clean
+    apt-get install -y clang cuda-compiler-12-1 cuda-nsight-systems-12-1 && \
+    apt-get -y clean && \
+    update-alternatives --set cc /usr/bin/clang && \
+    update-alternatives --set c++ /usr/bin/clang++
 
 RUN chgrp -R 0 /opt/conda && \
     chmod -R a+w /opt/conda && \