机器学习-搭建环境-5：安装CUDA和cuDNN

一、安装CUDA10

在计算机上使用CUDA，我们需要如下的基本组件：

• 支持CUDA的GPU
• 一个CUDA支持的Linux版本和一个gcc编译器和相关工具链
• NVIDIA CUDA Toolkit

（1）检查是否安装了GPU

# 检查是否安装了GPU
lspci|grep -i nvidia

# 检查tensorflow是否支持GPU
$ python
>> import tensorflow as tf
>> sess = tf.Session(config=tf.ConfigProto(log_device_placement=True))

返回结果：

在进行CentOS最小化安装后，会发现lspci命令无法使用（commend not found），是因为相应的软件包没有安装。解决办法：

yum wharprovides */lspci
yum install pciutils

查看系统发行版本，CUDA支持的系统版本参考：