linux安装tensorflow-gpu记录

1.查看linux系统版本

 $cat /etc/redhat-release

2.查看nvida gpu版本

3.确认是否安装gcc

输入$gcc --version

4.下载CUDA地址：https://developer.nvidia.com/cuda-toolkit-archive根据Linux系统版本和nvida版本选择相应的CUDA安装

本文中下CUDA11.1.0 centos7版本

选择在线安装根据图中提示输入命令

wget https://developer.download.nvidia.com/compute/cuda/11.1.0/local_installers/cuda_11.1.0_455.23.05_linux.run

下载完成后

sudo sh cuda_11.1.0_455.23.05_linux.run

显示以下内容

输入accept按enter键显示以下内容，

只保留CUDA Toolkit选项其他可以去掉

设置option设置安装路径等

更改Library install path (不改不行，它会偷偷写入/var/lib)

设置好后选择install

添加环境变量 编辑vim ~/.bashrc

输入nvcc -V验证

安装cudnn下载配置

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

解压tar -zxvf cudnn-11.2-linux-x64-v8.1.1.33.tgz        

复制以下文件到CUDA安装目录下

修改权限

chmod a+r ~/usr/local/cuda-11.1/include/cudnn.h ~/usr/local/cuda-11.1/lib64/libcudnn*