ubuntu16.04+cuda10.0+cudnn7.6+tensorflow_gpu-1.11.0环境安装

slender-

已于 2023-02-16 11:31:47 修改

阅读量796

点赞数

分类专栏：视觉语义SLAM 文章标签： tensorflow 深度学习 python

于 2022-11-21 16:54:54 首次发布

本文链接：https://blog.csdn.net/slender_1031/article/details/127411613

版权

视觉语义SLAM 专栏收录该内容

4 篇文章 0 订阅

订阅专栏

为了搭深度学习环境,又装了一遍各种库,在此记录安装版本和流程.

ubuntu16.04+cuda10.0+cudnn7.6+tensorflow_gpu-1.11.0环境安装

1 安装NVIDIA显卡
2 安装CUDA10.0
3 安装CUDNN
4 安装tensorflow_gpu-1.11.0

1 安装NVIDIA显卡

查看自己的显卡型号:

lspci |grep -i nvidia

在这里插入图片描述

nvidia官网:https://www.nvidia.cn/geforce/drivers/
搜索对应驱动
在这里插入图片描述

修改/etc/modprobe.d/blacklist.conf,添加下面内容

sudo gedit /etc/modprobe.d/blacklist.conf

blacklist vga16fb
blacklist nouveau
blacklist rivafb
blacklist rivatv
blacklist nvidiafb

新建blacklist-nouveau.conf文件,添加下面内容

sudo gedit /etc/modprobe.d/blacklist-nouveau.conf

blacklist nouveau
blacklist lbm-nouveau
options nouveau modeset=0
alias nouveau off
alias lbm-nouveau off

sudo update-initramfs -u
reboot

查看nouveau模块是否被加载。若没输出，则执行下一步。

lsmod | grep nouveau

安装

sudo service lightdm stop
sudo ./NVIDIA-Linux-x86_64-xxx.run --no-opengl-files
sudo service lightdm start

nvidia-smi

在这里插入图片描述

2 安装CUDA10.0

下载
https://link.csdn.net/?target=https%3A%2F%2Fdeveloper.nvidia.com%2Fcuda-10.0-download-archive
在这里插入图片描述

sudo gedit ~/.bashrc

# for CUDA 10.0
export CUDA_10_0_HOME=/usr/local/cuda-10.0
export PATH=$PATH:$CUDA_10_0_HOME/bin 
export LD_LIBRARY_PATH=CUDA_10_0_HOME/lib64${LD_LIBRARY_PATH:+:${LD_LIBRARY_PATH}}

3 安装CUDNN

https://developer.nvidia.com/rdp/cudnn-archive

sudo dpkg -i libcudnn7_7.6.0.64-1+cuda10.0_amd64.deb 
sudo dpkg -i libcudnn7-dev_7.6.0.64-1+cuda10.0_amd64.deb 
cat /usr/local/cuda/include/cudnn.h | grep CUDNN_MAJOR -A 2