MoneyBall
MoneyBall

Reputation: 2573

downgrade tensorflow to use cuda 8

When I run tensorflow I get:

ImportError: libcublas.so.9.0: cannot open shared object file: No such file or directory

So I'd like to downgrade my tensorflow-gpu to use cuda 8. How do I do that with pip and which version should I download?

Upvotes: 3

Views: 5915

Answers (1)

layog
layog

Reputation: 4811

Use the 1.4.1 version of tensorflow as

pip install tensorflow-gpu==1.4.1

Upvotes: 12

Related Questions