Before you install Tensorflow on your Mac or your AWS, you must know !!

Recently I started playing Tensorflow on my mac, it very fun. But I found there are some traps about Tensorflow on Mac and AWS, so tell you know.

  1. Tensorflow is very slow.
  2. Now Tensorflow doesn’t support CUDA on Mac officially, but you can easy find a way to compile it to support CUDA. But almost all new Mac don’t contain a NVIDIA display card, so you still can’t use CUDA. OpenCL support is on the way, open to contributions, but how knows.
  3. The HPC of AWS ( g2.2xlarge
    and g2.8xlarge ) contain NVIDIA display card, but the card is blow the requirement of default CUDA setting of Tensorflow, so you can’t user precompiled Tensorflow, you need compile it by yourself, and set compute capability is 3.0. And there is a AMI (Amazon Linux x64 AMI with TensorFlow (GPU)), already include Tensorflow on AWA’ AMI market, but this AMI is using precompiled Tensorflow, so it is useless, you can’t use the GPU. And even after your compile and install you own Tensorflow, there is a bug of NVIDIA display driver card, you must update NVIDIA display card driver, and finally you can use GPU from the HPC.

4 Comments

  1. Mikesays:

    Have you found a way to use a cloud service for tensorflow that uses the GPU?

    • tinyfoolsays:

      As I said, AWS is OK, but you need compile it by yourself, and set compute capability is 3.0. and you must update NVIDIA display card driver.

    • In case you are still looking for AMIs to get started with TensorFlow and GPUs quickly on AWS, we maintain several deep learning AMIs in the AWS Marketplace and make sure they work. We just recently updated our TensorFlow AMI to 0.9 and added cuDNN 5 support.

      http://www.bitfusion.io/boost-machine-images/

  2. Hi, thanks for your notes! I, like others, have struggled in installing and making TensorFlow work. It took a lot of time and there were a multitude of errors. I created an AMI that is actively maintained and open-source. I hope the community can maintain one AMI where everyone can use in any region on any GPU instance. I mean, even the deep learning AMI supported by Amazon does not have TensorFlow working out of the box. I tested it on all g2 and p2 instances and you can start working on TensorFlow in less than 5 minutes by using this AMI. There’s absolutely 0 configuration needed.

    TFAMI: A TensorFlow Amazon Web Service (AWS) AMI that is open, free and works.
    https://github.com/ritchieng/tensorflow-aws-ami

Leave a comment

Please be polite. We appreciate that. Your email address will not be published and required fields are marked

To create code blocks or other preformatted text, indent by four spaces:

    This will be displayed in a monospaced font. The first four 
    spaces will be stripped off, but all other whitespace
    will be preserved.
    
    Markdown is turned off in code blocks:
     [This is not a link](http://example.com)

To create not a block, but an inline code span, use backticks:

Here is some inline `code`.

For more help see http://daringfireball.net/projects/markdown/syntax