Ubuntu16安装TensorFlow GPU版本

总体安装步骤

  • python
  • tensorflow
  • Nvidia显卡驱动
  • cuda
  • cudnn

注意事项

  • 安装完tensorflow之后直接尝试import tensorflow,并查看报错信息。其中包含了缺少的CUDA版本信息
  • 根据上一步下载对应的CUDA,一般是8.0,千万不要下载9.0
  • 在下载页面下方有Legacy Release,点击之后可以下载8.0
  • 禁用nouveau的驱动
  • 安装cuda的时候,不需要安装cuda自带的显卡驱动

推荐阅读

ubuntu16 CUDA tensorflow

你可能感兴趣的:(Ubuntu16安装TensorFlow GPU版本)