install CUDA Toolkit v9.0 and cuDNN v7.0 on Ubuntu 16.04

From:https://gist.github.com/ashokpant/5c4e9481615f54af4025ab2085f85869
Thanks.

#!/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-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}
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= P A T H : / u s r / l o c a l / c u d a − 9.0 / b i n e x p o r t C U D A H O M E = {PATH}:/usr/local/cuda-9.0/bin export CUDA_HOME= PATH:/usr/local/cuda9.0/binexportCUDAHOME={CUDA_HOME}:/usr/local/cuda:/usr/local/cuda-9.0
export LD_LIBRARY_PATH= L D L I B R A R Y P A T H : / u s r / l o c a l / c u d a − 9.0 / l i b 64 e x p o r t L D L I B R A R Y P A T H = {LD_LIBRARY_PATH}:/usr/local/cuda-9.0/lib64 export LD_LIBRARY_PATH= LDLIBRARYPATH:/usr/local/cuda9.0/lib64exportLDLIBRARYPATH=LD_LIBRARY_PATH:/usr/local/cuda/extras/CUPTI/lib64

  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值