ubuntu18.04安装cuda9.2、cudnn7.1

0.系统版本

ubuntu18.04,本次安装并未降级gcc,仍然是系统自带的gcc-7, 重中之重在于先正确安装显卡驱动:
说明:如果不想源码安装tf,安装9.0cuda即可

# 卸载可能存在的nvidia驱动
sudo apt-get remove --purge nvidia*
​
#把 nouveau 驱动加入黑名单并禁用用 nouveau 内核模块
sudo nano /etc/modprobe.d/blacklist-nouveau.conf
​
# 在文件 blacklist-nouveau.conf 中加入如下内容:
blacklist nouveau
options nouveau modeset=0# 保存退出,执行
sudo update-initramfs -u
​
# 添加源
sudo add-apt-repository ppa:graphics-drivers/ppa
sudo apt update
# 下面这一步网络比较慢,建议用proxychains
sudo apt install nvidia-driver-396
之后重启

1. cuda9

# 下载.run文件
sudo sh .run
​
# 选择不安装驱动,因为已经安装过了,用这里的驱动安装会有问题
# 最后要设置环境变量(安装结束的时候也会提示)# 编辑 ~/.bashrc文件,在文末追加export PATH=/usr/local/cuda/bin:$PATH
export LD_LIBRARY_PATH=/usr/local/cuda/extras/CUPTI/lib64:/usr/local/cuda/lib64:$LD_LIBRARY_PATH
export CUDA_HOME=/usr/local/cuda
​
# 使配置立即生效
source ~/.bashrc
安装:
sudo apt-get install libcupti-dev

2. cudnn

首先登录nvidiag官网,下载相应文件( cudnn-9.2-linux-x64-v7.1.tgz)

tar -zxvf cudnn-9.2-linux-x64-v7.1.tgz
sudo cp cuda/include/cudnn.h /usr/local/cuda/include/
sudo cp cuda/lib64/libcudnn* /usr/local/cuda/lib64/ -d
sudo chmod a+r /usr/local/cuda/include/cudnn.h
sudo chmod a+r /usr/local/cuda/lib64/libcudnn*

备注
源码编译 tf, 参考tensorflow.org的install from source

你可能感兴趣的:(环境配置)