git.iko.soy/stackage
on master help sign in
update to cuda-10
6d54752Trevor L. McDonell
Dockerfile+3−3
compare with master
unifiedsplit
@@ -4,9 +4,9 @@ ENV HOME /home/stackage
4ENV LANG en_US.UTF-8
4ENV LANG en_US.UTF-8
5
5
6# NOTE: also update debian-bootstrap.sh when cuda version changes
6# NOTE: also update debian-bootstrap.sh when cuda version changes
7−ENV PATH /usr/local/cuda-8.0/bin:/opt/ghc/8.6.1/bin:/usr/sbin:/usr/bin:/sbin:/bin
7+ENV PATH /usr/local/cuda-10.0/bin:/opt/ghc/8.6.1/bin:/usr/sbin:/usr/bin:/sbin:/bin
8−ENV CUDA_PATH /usr/local/cuda-8.0
8+ENV CUDA_PATH /usr/local/cuda-10.0
9−ENV LD_LIBRARY_PATH=/usr/local/cuda-8.0/lib64:/usr/local/cuda-8.0/nvvm/lib64
9+ENV LD_LIBRARY_PATH=/usr/local/cuda-10.0/lib64:/usr/local/cuda-10.0/nvvm/lib64
10
10
11ADD debian-bootstrap.sh /tmp/debian-bootstrap.sh
11ADD debian-bootstrap.sh /tmp/debian-bootstrap.sh
12RUN /tmp/debian-bootstrap.sh && rm /tmp/debian-bootstrap.sh
12RUN /tmp/debian-bootstrap.sh && rm /tmp/debian-bootstrap.sh
debian-bootstrap.sh+4−3
@@ -268,12 +268,13 @@ cd /tmp \
268# NOTE: also update Dockerfile when cuda version changes
268# NOTE: also update Dockerfile when cuda version changes
269# Install CUDA toolkit
269# Install CUDA toolkit
270# The current version can be found at: https://developer.nvidia.com/cuda-downloads
270# The current version can be found at: https://developer.nvidia.com/cuda-downloads
271−CUDA_PKG=8.0.61-1 # update this on new version
271+CUDA_PKG=10.0.130-1
272−CUDA_VER=${CUDA_PKG:0:3}
272+CUDA_VER=10.0
273−CUDA_APT=${CUDA_VER/./-}
273+CUDA_APT=10-0
274
274
275pushd /tmp \
275pushd /tmp \
276 && wget https://developer.download.nvidia.com/compute/cuda/repos/ubuntu1604/x86_64/cuda-repo-ubuntu1604_${CUDA_PKG}_amd64.deb \
276 && wget https://developer.download.nvidia.com/compute/cuda/repos/ubuntu1604/x86_64/cuda-repo-ubuntu1604_${CUDA_PKG}_amd64.deb \
277+ && apt-key adv --fetch-keys http://developer.download.nvidia.com/compute/cuda/repos/ubuntu1604/x86_64/7fa2af80.pub \
277 && dpkg -i cuda-repo-ubuntu1604_${CUDA_PKG}_amd64.deb \
278 && dpkg -i cuda-repo-ubuntu1604_${CUDA_PKG}_amd64.deb \
278 && apt-get update -qq \
279 && apt-get update -qq \
279 && apt-get install -y cuda-drivers cuda-core-${CUDA_APT} cuda-cudart-dev-${CUDA_APT} cuda-cufft-dev-${CUDA_APT} cuda-cublas-dev-${CUDA_APT} cuda-cusparse-dev-${CUDA_APT} cuda-cusolver-dev-${CUDA_APT} \
280 && apt-get install -y cuda-drivers cuda-core-${CUDA_APT} cuda-cudart-dev-${CUDA_APT} cuda-cufft-dev-${CUDA_APT} cuda-cublas-dev-${CUDA_APT} cuda-cusparse-dev-${CUDA_APT} cuda-cusolver-dev-${CUDA_APT} \