抱歉,您的浏览器无法访问本站

本页面需要浏览器支持(启用)JavaScript


了解详情 >

TensorFlow For Docker 初体验


TensorFlow 是一套开源的机器学习工具。一般来说只用 TensorFlow 的话配置运行环境什么的并没有特别坑的地方,但如果想用到 GPU 加速计算的话配置起来就要费好大一番力气了,还经常遇到各种版本不兼容、找不到依赖关系等问题,让人头疼。而 Docker 刚好是解决开源软件各种依赖关系的神物,NVIDIA 刚好又有工具能让容器用上 GPU 进行计算。

下面分享配置过程。

测试环境是 GTX850M+Ubuntu16.04.3+CUDA9.0+GeForce 384.00

驱动及 CUDA 安装过程参考即将到来的另一篇文章。