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
我这里装的是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=
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
P
A
T
H
=
LD_LIBRARY_PATH:/usr/local/cuda-9.0/lib64 export PATH=
LDLIBRARYPATH:/usr/local/cuda−9.0/lib64exportPATH=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)删除文件