tensorflow安装的一些记录

安装教程：(13条消息) Linux下安装tensorflow-gpu,cuda及cudnn_zjy1125的博客-CSDN博客

 

tensorflow和cuda的版本对应关系：https://tensorflow.google.cn/install/source_windows?hl=en#gpu

 

 

 

将CUDNN的.tgz压缩包解压：

tar -xzvf cudnn-10.0-linux-x64-v7.5.1.10.tgz

解压后的操作：(解压后的文件名就叫cuda)

cp /home/dailh/cuda/include/cudnn.h /home/dailh/cuda-10.0/include/
cp /home/dailh/cuda/lib64/libcudnn* /home/dailh/cuda-10.0/lib64

chmod a+r /home/dailh/cuda-10.0/include/cudnn.h /home/dailh/cuda-10.0/lib64/libcudnn*

 

在设置~/.bashrc的时候都可以设置到：

export LD_LIBRARY_PATH=$LD_LIBRARY_PATH:/usr/local/cuda/lib64
export PATH=$PATH:/usr/local/cuda/bin
export CUDA_HOME=$CUDA_HOME:/usr/local/cuda

而不是设置到/usr/local/cuda-10.1

可以设置/usr/local/cuda的软链接到不同的cuda安装路径来使用不同的cuda version：

命令行输入以下命令，断开cuda文件与cuda-10.1的符号链接：

sudo rm -rf cuda

命令行输入以下命令，建立cuda文件与cuda-9.0的符号链接：

sudo ln -s /usr/local/cuda-9.0 /usr/local/cuda

 可见：(13条消息) Linux下安装cuda和对应版本的cudnn_水哥很水的博客-CSDN博客_linux安装cuda和cudnn

 

 

 

查看CUDNN的版本： vim /usr/local/cuda/include/cudnn.h :

#define CUDNN_MAJOR 7
#define CUDNN_MINOR 6
#define CUDNN_PATCHLEVEL 1

#define CUDNN_VERSION (CUDNN_MAJOR * 1000 + CUDNN_MINOR * 100 + CUDNN_PATCHLEVEL)

 说明CUDNN的版本是7.601

 

 

测试TF是否能使用GPU：

tf.config.list_physical_devices('GPU')

老版本使用：

tf.test.is_gpu_available()

返回True或者False