Pierre Carceller
Pierre Carceller

Reputation: 35

tensorflow compiled with cuda 9.1

I'm beginer on tensorflow and i want to install the gpu version with cuda 9.0. The problem is that my gpu (Nvidia mx150) doesn't work with cuda 9.0 (only with cuda 9.1)... So i tried to compile my own version of tensorflow with cuda 9.1. But I'm still blocked cause during the compilation the compiler find an error so it doesn't compile... I don't know why but it's very frustrating. So if you have an already compiled version of tensorflow with cuda 9.1 I'm very intrested !

thanks in advance !

Upvotes: 0

Views: 672

Answers (2)

Boyang
Boyang

Reputation: 103

I had the exact same problem a few weeks ago.

The problem is that the current version of TensorFlow (1.7) does not support CUDA9.1, please check this issue comment and the discussion below.

Here are some options that I found:

  1. Compile TensorFlow from the source code by your own
  2. Find an existing whl file. ex. I fixed the issue by using a whl from the repo (same as Y. Luo's answer)
  3. Downgrade to CUDA 9.0
  4. Wait until Tensorflow supports CUDA9.1 :)

Upvotes: 1

Y. Luo
Y. Luo

Reputation: 5732

If you don't have to use tensorflow 1.7, this repository might have what you want. Just to be clear, I never tried any of them myself.

If you need to install on Windows, this repository might be helpful.

Upvotes: 0

Related Questions