e2c05ba · stackage · git.iko.soy

git.iko.soy/stackage

Bake cuda env vars into Dockerfile

commite2c05bab2bd0397e0a65eb272ad34fa626b88329 authorDan Burton <danburton.email@gmail.com> dateFri 1 Sep 2017, 12:12 (9 years ago) parente1d31dd

2 files changed+6−1

unifiedsplit

Dockerfile+5−1

@@ -2,7 +2,11 @@ FROM fpco/pid1:16.04

33ENV HOME /home/stackage

44ENV LANG en_US.UTF-8

5−ENV PATH /opt/ghc/8.2.1/bin:/usr/sbin:/usr/bin:/sbin:/bin

6+# NOTE: also update debian-bootstrap.sh when cuda version changes

7+ENV PATH /usr/local/cuda-8.0/bin:/opt/ghc/8.2.1/bin:/usr/sbin:/usr/bin:/sbin:/bin

8+ENV CUDA_PATH /usr/local/cuda-8.0

9+ENV LD_LIBRARY_PATH=/usr/local/cuda-8.0/lib64:/usr/local/cuda-8.0/nvvm/lib64

610

711ADD debian-bootstrap.sh /tmp/debian-bootstrap.sh

812RUN /tmp/debian-bootstrap.sh && rm /tmp/debian-bootstrap.sh

debian-bootstrap.sh+1

@@ -207,6 +207,7 @@ curl https://storage.googleapis.com/tensorflow/libtensorflow/libtensorflow-cpu-l

207207 && rm libtensorflow.tar.gz \

208208 && ldconfig

209209

210+# NOTE: also update Dockerfile when cuda version changes

210211# Install CUDA toolkit

211212# The current version can be found at: https://developer.nvidia.com/cuda-downloads

212213CUDA_PKG=8.0.61-1 # update this on new version