安装CUDA和cuDNN

下面是CentOS系统安装过程

1、CUDA安装
官方下载:https://developer.nvidia.com/cuda-downloads

sudo rpm -i cuda-repo-rhel7-8.0.44-1.x86_64.rpm
sudo yum clean all
sudo yum install cuda

设置环境变量

# CUDA Toolkit
export LD_LIBRARY_PATH=/usr/local/cuda-7.5
/lib64:$LD_LIBRARY_PATH
export PATH=/usr/local/cuda-7.5
/bin:$PATH

检查是否安装成功

#输入以下命令,检查是否配置好。
nvcc --version

2、cuDNN安装

官网下载:https://developer.nvidia.com/rdp/cudnn-download

然后把lib文件加入到系统能找到的lib文件夹里, 把头文件加到系统能找到的include文件夹里就可以。这里把他们加到CUDA的文件夹下。

tar -zxf cudnn-7.5-linux-x64-v5.0-ga.tgz
cd cuda
sudo cp -P lib64/* /usr/local/cuda/lib64/
sudo cp -P include/* /usr/local/cuda/include/
sudo chmod a+r /usr/local/cuda/lib64/libcudnn*

注:-P参数是为了保留软链接

你可能感兴趣的:(Linux,深度学习,TensorFlow)