How to install Tensorflow with GPU support, OpenCV with Python3 on Amazon EC2 Instance

I made Github Gist with comments -> here

This setup is available as a public AMI in US-East (N. Virginia): ami-9d0f3ff7

Current image configuration:

  • Python: 3.4
  • OpenCV: 3.1
  • Tensorflow: 0.7
  • nvidia Cuda: 7.0
  • cuDNN lib: 6.5
  • Bazel: 1.0.5

Important notes:

  • You have to register and download CUDNN library beforehand. Registration approval may take up to some days!
  • Installing OpenCV verify WITH_CUDA flag. It must be FALSE, so you can generate cv2.so normally. OpenCV with Python does not support GPU now, you have to write your own wrapper for that.
  • Whole setup will take 1.5+ hours on g2.2xlarge instance.

results matching ""

    No results matching ""