haskell/stackage
1
0
Fork 0
mirror of https://github.com/commercialhaskell/stackage.git synced 2026-02-13 14:57:30 +01:00
Code Issues Actions Packages Projects Releases Wiki Activity

Bake cuda env vars into Dockerfile

Browse Source
This commit is contained in:
Dan Burton 2017-09-01 12:12:59 -07:00
parent e1d31dd044
commit e2c05bab2b
No known key found for this signature in database
GPG Key ID: C11E00118F0CE509
2 changed files with 6 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
Dockerfile
View File

@ -2,7 +2,11 @@ FROM fpco/pid1:16.04
ENV HOME /home/stackage ENV HOME /home/stackage
ENV LANG en_US.UTF-8 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 ADD debian-bootstrap.sh /tmp/debian-bootstrap.sh
RUN /tmp/debian-bootstrap.sh && rm /tmp/debian-bootstrap.sh RUN /tmp/debian-bootstrap.sh && rm /tmp/debian-bootstrap.sh

1
debian-bootstrap.sh
View File

@ -207,6 +207,7 @@ curl https://storage.googleapis.com/tensorflow/libtensorflow/libtensorflow-cpu-l
&& rm libtensorflow.tar.gz \ && rm libtensorflow.tar.gz \
&& ldconfig && ldconfig
# NOTE: also update Dockerfile when cuda version changes
# Install CUDA toolkit # Install CUDA toolkit
# The current version can be found at: https://developer.nvidia.com/cuda-downloads # The current version can be found at: https://developer.nvidia.com/cuda-downloads
CUDA_PKG=8.0.61-1 # update this on new version CUDA_PKG=8.0.61-1 # update this on new version