From e2c05bab2bd0397e0a65eb272ad34fa626b88329 Mon Sep 17 00:00:00 2001 From: Dan Burton Date: Fri, 1 Sep 2017 12:12:59 -0700 Subject: [PATCH] Bake cuda env vars into Dockerfile --- Dockerfile | 6 +++++- debian-bootstrap.sh | 1 + 2 files changed, 6 insertions(+), 1 deletion(-) diff --git a/Dockerfile b/Dockerfile index 56076512..710e7bbe 100644 --- a/Dockerfile +++ b/Dockerfile @@ -2,7 +2,11 @@ FROM fpco/pid1:16.04 ENV HOME /home/stackage ENV LANG en_US.UTF-8 -ENV PATH /opt/ghc/8.2.1/bin:/usr/sbin:/usr/bin:/sbin:/bin + +# NOTE: also update debian-bootstrap.sh when cuda version changes +ENV PATH /usr/local/cuda-8.0/bin:/opt/ghc/8.2.1/bin:/usr/sbin:/usr/bin:/sbin:/bin +ENV CUDA_PATH /usr/local/cuda-8.0 +ENV LD_LIBRARY_PATH=/usr/local/cuda-8.0/lib64:/usr/local/cuda-8.0/nvvm/lib64 ADD debian-bootstrap.sh /tmp/debian-bootstrap.sh RUN /tmp/debian-bootstrap.sh && rm /tmp/debian-bootstrap.sh diff --git a/debian-bootstrap.sh b/debian-bootstrap.sh index 9c59f1d0..d294a690 100755 --- a/debian-bootstrap.sh +++ b/debian-bootstrap.sh @@ -207,6 +207,7 @@ curl https://storage.googleapis.com/tensorflow/libtensorflow/libtensorflow-cpu-l && rm libtensorflow.tar.gz \ && ldconfig +# NOTE: also update Dockerfile when cuda version changes # Install CUDA toolkit # The current version can be found at: https://developer.nvidia.com/cuda-downloads CUDA_PKG=8.0.61-1 # update this on new version