TensorFlow 是一套开源的机器学习工具。一般来说只用 TensorFlow 的话配置运行环境什么的并没有特别坑的地方,但如果想用到 GPU 加速计算的话配置起来就要费好大一番力气了,还经常遇到各种版本不兼容、找不到依赖关系等问题,让人头疼。而 Docker 刚好是解决开源软件各种依赖关系的神物,NVIDIA 刚好又有工具能让容器用上 GPU 进行计算。
下面分享配置过程。
测试环境是 GTX850M+Ubuntu16.04.3+CUDA9.0+GeForce 384.00
驱动及 CUDA 安装过程参考即将到来的另一篇文章。