1. ubuntu16.04
2. NVIDIA驱动版本 387(
查看GPU型号
lspci | grep -i
vga
如果nvidia-smi显示无command或者无法连接driver
先卸载:
sudo apt-get remove --purge nvidia-*
再直接采用software update更新)
3. CUDA8.0(deviceQuery)
cd /usr/local/cuda/samples/1_Utilities/deviceQuery
sudo deviceQuery
如果能显示则不用重新安装
4. cuDNN5.1
5. tensorflow1.2.0(只有1.2.0版本配cuDNN5.1)
如果运行时没有GPU信息出现,则先
sudo pip uninstall tensorflow (卸载CPU版本tensorflow)
sudo pip uninstall tensorflow-gpu (卸载GPU版本tensorflow)
sudo pip install tensorflow-gpu==1.2 (安装GPU版本tensorflow-gpu-1.2.0)