Ubuntu快速运行Tensorflow

Ubuntu快速运行Tensorflow

最近入手了一块Titan V,自然是要玩一玩机器学习。这块卡不仅CUDA核达到了5120个,还有了640个Tensor计算核,安装TesnsorFlow是首选了。

我的硬件环境是:CPU-7980EX,内存-海盗船16G*8,显卡-Titan V。

安装的软件包括:

  1. Ubuntu16.04 LTS
  2. Nvidia Driver 396.42
  3. Nvidia CUDA 9.1
  4. Nvidia cuDNN 7.1.3
  5. Nvidia cuTensor 4.0
  6. Docker CE 18.03
  7. Docker-Nvidia 2.0.3
  8. Tensorflow for Docker 1.8

硬件、驱动和软件都是最新的版本,中间踩了不少坑。记录下来,供参考。

1、安装 Ubuntu16.04 LTS

2、安装Nvidia图形驱动396

3、安装CUDA 9.1

4、安装cuDNN 7.1.3

5、安装cuTensor 4.0

6、安装Docker CE 18.03

7、安装Docker-Nvidia

8、安装Tensorflow for Docker

9、运行Jupyter with Tensorflow

猜你喜欢

转载自my.oschina.net/u/2306127/blog/1807648
今日推荐