git.iko.soy/stackage
on master help sign in
Bake cuda env vars into Dockerfile
e2c05baDan Burton
Dockerfile+5−1
compare with master
unifiedsplit
@@ -2,7 +2,11 @@ FROM fpco/pid1:16.04
2
2
3ENV HOME /home/stackage
3ENV HOME /home/stackage
4ENV LANG en_US.UTF-8
4ENV LANG en_US.UTF-8
5+
6+# NOTE: also update debian-bootstrap.sh when cuda version changes
5−ENV PATH /opt/ghc/8.2.1/bin:/usr/sbin:/usr/bin:/sbin:/bin
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
6
10
7ADD debian-bootstrap.sh /tmp/debian-bootstrap.sh
11ADD debian-bootstrap.sh /tmp/debian-bootstrap.sh
8RUN /tmp/debian-bootstrap.sh && rm /tmp/debian-bootstrap.sh
12RUN /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
207 && rm libtensorflow.tar.gz \
207 && rm libtensorflow.tar.gz \
208 && ldconfig
208 && ldconfig
209
209
210+# NOTE: also update Dockerfile when cuda version changes
210# Install CUDA toolkit
211# Install CUDA toolkit
211# The current version can be found at: https://developer.nvidia.com/cuda-downloads
212# The current version can be found at: https://developer.nvidia.com/cuda-downloads
212CUDA_PKG=8.0.61-1 # update this on new version
213CUDA_PKG=8.0.61-1 # update this on new version