Skip to content

Instantly share code, notes, and snippets.

@marks

marks/gist:c8b308b65723a15559cf136a945100f4

Created January 5, 2017 19:29
Show Gist options
  • Star 0 You must be signed in to star a gist
  • Fork 0 You must be signed in to fork a gist
  • Save marks/c8b308b65723a15559cf136a945100f4 to your computer and use it in GitHub Desktop.
Save marks/c8b308b65723a15559cf136a945100f4 to your computer and use it in GitHub Desktop.
Download ZIP
Raw
gistfile1.txt
RUN \
export CUDA_HOME=/usr/local/cuda-7.5 && \
export CUDA_ROOT="/usr/local/cuda-7.5/bin" && \
export PATH=/usr/local/cuda-7.5/bin:$PATH && \
export LD_LIBRARY_PATH=/usr/local/cuda-7.5/lib64 && \
cd ~ && \
wget https://app.dominodatalab.com/johnjoo/cuDNN_for_TF/raw/latest/cudnn-7.5-linux-x64-v5.1.tgz -O cudnn-7.5-linux-x64-v5.1.tgz.gz && \
gunzip cudnn-7.5-linux-x64-v5.1.tgz.gz && \
tar xvzf cudnn-7.5-linux-x64-v5.1.tgz && \
cp cuda/include/cudnn.h /usr/local/cuda-7.5/include && \
cp cuda/lib64/libcudnn* /usr/local/cuda-7.5/lib64 && \
chmod a+x /usr/local/cuda-7.5/include/cudnn.h /usr/local/cuda-7.5/lib64/libcudnn* && \
rm -rf cuda && \
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment