一、查看机器上navidia显卡信息
lspci | grep -i nvidia
lspci | grep -i vga
如果结果显示有nvidia字样还带有型号啥的,说明你机器上有nvidia显卡,可能能够使用GPU(这里用可能二字是因为计算是nvidia显卡也不一定是能用GPU加速计算,对于TensorFlow而言,应该是算力达到一定程度才可以),若果要是输入刚才命令啥也没有显示,而显示:
00:02.0 VGA compatible controller: Cirrus Logic GD 5446
说明机器上没有nvidia显卡,只有一个普通显卡,即便你安装cuda也没用,不能加速计算。
二、确认Linux版本是否支持cuda
uname -m && cat /etc/*release
若有信息输出,说明linux上是支持
三、检查gcc是否安装
gcc --version
若没有安装,可以使用一下命令安装。
sudo yum install gcc gcc-c++
四、安装python
wget http://repo.continuum.io/archive/Anaconda3-5.1.0-Linux-x86_64.sh
五、安装TensorFlow-gpu
pip --default-timeout=1000 install -