kfac icon indicating copy to clipboard operation
kfac copied to clipboard
verified publisher icon tensorflow

What versions of Python and TensorFlow are required for using K-FAC?

Open johannafrost opened this issue 5 years ago • 6 comments

I am trying to install and import K-FAC without success. I am using Anaconda, python 3.7.6 and have tried TensorFlow 2.1 and 1.15. Someone told me tf 1.5 might work, but I can't install that version via pip (by the error message, it seems like 1.13.0 is the oldest version I can install). Any ideas? Thanks!

johannafrost avatar Mar 11 '20 21:03 johannafrost

I am also running into this issue. @johannafrost have you had any success?

ddusseau avatar May 04 '20 13:05 ddusseau

@ddusseau tensorflow 1.15 and tensorflow-probability 0.8 works for me.

johannafrost avatar May 04 '20 17:05 johannafrost

I had tensorflow-probability 0.9. I switched to 0.8 and it worked. Thank you!

ddusseau avatar May 04 '20 22:05 ddusseau

I think that newer TF libraries should work with K-FAC, but only if your user code runs in TF1 legacy mode. Actual TF2 behavior is not supported by K-FAC.

james-martens avatar May 30 '20 22:05 james-martens

We had a proposal for a TF2 porting at https://github.com/tensorflow/addons/issues/2388. It is under an TF ecosystem review.

bhack avatar Feb 12 '21 13:02 bhack

on Google Colab I use KFAC with:

%tensorflow_version 1.x

and so far no problems. I tried with newer versions, no success.

voaneves avatar Feb 19 '21 16:02 voaneves