set reasonable CUDA_HOME default

ucsd-ets · Sep 12, 2023 · 51b260c · 51b260c
1 parent 2e2a821
commit 51b260c
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/Dockerfile b/Dockerfile
@@ -43,6 +43,8 @@ RUN echo '/usr/local/cuda/targets/x86_64-linux/lib' >> /etc/ld.so.conf.d/000_cud
     ( echo '/usr/local/nvidia/lib'; echo '/usr/local/nvidia/lib64' ) >> /etc/ld.so.conf.d/nvidia.conf && \
     ldconfig 
 
+ENV CUDA_HOME=/usr/local/cuda
+
 ###########################################
 # Remainder of install as nonprivleged user