tensorflow gpu安装

1.cuda

按官网命令

https://developer.nvidia.com/cuda-90-download-archive?target_os=Linux&target_arch=x86_64&target_distro=CentOS&target_version=7&target_type=rpmnetwork

更改/usr/local 软链接cuda指向新路径 

查看版本 cat /usr/local/cuda/version.txt 

2.cudnn

下载 https://developer.nvidia.com/cudnn

解压

$ cp cudnn-8.0-linux-x64-v5.1.solitairetheme8 cudnn-8.0-linux-x64-v5.1.tgz

$ tar -xvf cudnn-8.0-linux-x64-v5.1.tgz

安装

https://blog.csdn.net/yaoqiwaimai/article/details/78594154

查看 CUDNN 版本：

cat /usr/local/cuda/include/cudnn.h | grep CUDNN_MAJOR -A 2