Ubuntu快速运行Tensorflow
最近入手了一块Titan V,自然是要玩一玩机器学习。这块卡不仅CUDA核达到了5120个,还有了640个Tensor计算核,安装TesnsorFlow是首选了。
我的硬件环境是:CPU-7980EX,内存-海盗船16G*8,显卡-Titan V。
安装的软件包括:
- Ubuntu16.04 LTS
- Nvidia Driver 396.42
- Nvidia CUDA 9.1
- Nvidia cuDNN 7.1.3
- Nvidia cuTensor 4.0
- Docker CE 18.03
- Docker-Nvidia 2.0.3
- Tensorflow for Docker 1.8
硬件、驱动和软件都是最新的版本,中间踩了不少坑。记录下来,供参考。