1.查看驱动版本:
cat /proc/driver/nvidia/version
可以先查看下有没有装驱动:nvidia-smi

2.查看tensorflow与cuda、cudnn对应版本:
检查驱动没有问题后,装cuda前先要看tensorflow的对应关系。
https://tensorflow.google.cn/install/source_windows #cpu
https://tensorflow.google.cn/install/source #linux
ubuntu16.04 cuda9 cudnn7 tensorflow-gpu1.11.0 2019.11亲测有效
我这里装的是tensorflow1.11.0

3.cuda安装
(1)下载地址:https://developer.nvidia.com/cuda-toolkit-archive
(2)sudo sh cuda_10.0.130_410.48_linux.run
(3)accept
n(driver)
y
y
y
(4)打开主目录下的 .bashrc文件添加如下路径,按Ctrl+H键显示隐藏文件。
export LD_LIBRARY_PATH=LDLIBRARYPATH:/usr/local/cuda9.0/lib64exportPATH=LD_LIBRARY_PATH:/usr/local/cuda-9.0/lib64 export PATH=PATH:/usr/local/cuda-9.0/bin
export CUDA_HOME=$CUDA_HOME:/usr/local/cuda-9.0
(5)source ~/.bashrc
(6)检查:nvcc --version

4.cudnn安装
(1)下载地址:https://developer.nvidia.com/rdp/cudnn-archive
(2)sudo cp cuda/include/cudnn.h /usr/local/cuda/include/
sudo cp cuda/lib64/libcudnn* /usr/local/cuda/lib64/
sudo chmod a+r /usr/local/cuda/include/cudnn.h
sudo chmod a+r /usr/local/cuda/lib64/libcudnn*
(3)查看版本:cat /usr/local/cuda/include/cudnn.h | grep CUDNN_MAJOR -A 2

5.conda卸载东西:
安装tensorflow新版本把之前安装的tensorflow版本卸载掉(conda安装的)
conda list
conda uninstall xxx

6.conda安装tensorflow
(1)查找合适的源:anaconda search -t conda tensorflow-gpu
查找安装命令:anaconda show fwaters/tensorflow-gpu
安装 conda install --channel https://conda.anaconda.org/fwaters tensorflow-gpu
(2)直接安装想要的版本:conda install tensorflow-gpu=1.14.0
安装好之后可能会出现anaconda 自带的numpy版本过高,有warning。
卸载自带的numpy:pip uninstall numpy
安装16.0即可:pip install numpy==1.16.0

7.cuda卸载
(1)sudo /usr/local/cuda-10.0/bin/uninstall_cuda_10.0.pl
(2)删除文件夹

8.cudnn卸载
(1)sudo rm -rf /usr/local/cuda/lib64/libcudnn
(2)sudo rm -rf /usr/local/cuda/include/cudnn.h
(3)删除文件

ubuntu16.04 cuda9 cudnn7 tensorflow-gpu1.11.0 2019.11亲测有效

相关文章: