From 51b260cfb526ca573128d2cc0c04e6c880842d3d Mon Sep 17 00:00:00 2001 From: Adam Tilghman Date: Mon, 11 Sep 2023 18:58:56 -0700 Subject: [PATCH] set reasonable CUDA_HOME default --- Dockerfile | 2 ++ 1 file changed, 2 insertions(+) diff --git a/Dockerfile b/Dockerfile index 82d1694..95131f3 100644 --- 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