Skip to content

Instantly share code, notes, and snippets.

@gambitier

gambitier/cuda_9.0_cudnn_7.0.sh

Forked from ashokpant/cuda_9.0_cudnn_7.0.sh
Created August 13, 2018 18:27
Show Gist options
  • Save gambitier/455ee18aab4b085c51ccf9e73b51e0f1 to your computer and use it in GitHub Desktop.
Save gambitier/455ee18aab4b085c51ccf9e73b51e0f1 to your computer and use it in GitHub Desktop.
Download ZIP

Revisions

  1. @ashokpant ashokpant revised this gist Apr 17, 2018. 1 changed file with 1 addition and 1 deletion.
    2 changes: 1 addition & 1 deletion cuda_9.0_cudnn_7.0.sh
    View file
    Original file line number Diff line number Diff line change
    @@ -7,7 +7,7 @@ wget https://developer.nvidia.com/compute/cuda/9.0/Prod/local_installers/${CUDA_
    sudo dpkg -i ${CUDA_REPO_PKG}
    sudo apt-key adv --fetch-keys http://developer.download.nvidia.com/compute/cuda/repos/ubuntu1604/x86_64/7fa2af80.pub
    sudo apt-get update
    sudo apt-get -y install cuda
    sudo apt-get -y install cuda-9-0

    CUDA_PATCH1="cuda-repo-ubuntu1604-9-0-local-cublas-performance-update_1.0-1_amd64-deb"
    wget https://developer.nvidia.com/compute/cuda/9.0/Prod/patches/1/${CUDA_PATCH1}
  2. @ashokpant ashokpant created this gist Mar 5, 2018.
    30 changes: 30 additions & 0 deletions cuda_9.0_cudnn_7.0.sh
    View file
    Original file line number Diff line number Diff line change
    @@ -0,0 +1,30 @@
    #!/bin/bash

    # install CUDA Toolkit v9.0
    # instructions from https://developer.nvidia.com/cuda-downloads (linux -> x86_64 -> Ubuntu -> 16.04 -> deb)
    CUDA_REPO_PKG="cuda-repo-ubuntu1604-9-0-local_9.0.176-1_amd64-deb"
    wget https://developer.nvidia.com/compute/cuda/9.0/Prod/local_installers/${CUDA_REPO_PKG}
    sudo dpkg -i ${CUDA_REPO_PKG}
    sudo apt-key adv --fetch-keys http://developer.download.nvidia.com/compute/cuda/repos/ubuntu1604/x86_64/7fa2af80.pub
    sudo apt-get update
    sudo apt-get -y install cuda

    CUDA_PATCH1="cuda-repo-ubuntu1604-9-0-local-cublas-performance-update_1.0-1_amd64-deb"
    wget https://developer.nvidia.com/compute/cuda/9.0/Prod/patches/1/${CUDA_PATCH1}
    sudo dpkg -i ${CUDA_PATCH1}
    sudo apt-get update

    # install cuDNN v7.0
    CUDNN_PKG="libcudnn7_7.0.5.15-1+cuda9.0_amd64.deb"
    wget https://github.com/ashokpant/cudnn_archive/raw/master/v7.0/${CUDNN_PKG}
    sudo dpkg -i ${CUDNN_PKG}
    sudo apt-get update

    # install NVIDIA CUDA Profile Tools Interface ( libcupti-dev v9.0)
    sudo apt-get install cuda-command-line-tools-9-0

    # set environment variables
    export PATH=${PATH}:/usr/local/cuda-9.0/bin
    export CUDA_HOME=${CUDA_HOME}:/usr/local/cuda:/usr/local/cuda-9.0
    export LD_LIBRARY_PATH=${LD_LIBRARY_PATH}:/usr/local/cuda-9.0/lib64
    export LD_LIBRARY_PATH=$LD_LIBRARY_PATH:/usr/local/cuda/extras/CUPTI/lib64