Hello all! Anybody know of any good docker images that’ll run on a Xavier NX with a Keras frontend for TF? I note that NGC files don’t have keras in the docker files, due to its rapid changes and the fact that it can be a frontend for multiple things. I tried just running “pip3 install keras” in the m4L container (and it said it worked), but then started getting errors about having the wrong tensorflow installed when I wanted to import things. I just want one that works out of the box!
And yes, I know, training my own models on the xavier is going to be comparatively slow. It’s my only option right now, unless someone at NVidia has a spare 3080 laying around in the back I can buy! (My ancient geforce 760 won’t run modern tensorflow, haha). And yes, I see that pytorch is on plenty of the NGC docker files, but I’m more well-versed in keras! I could probably pick pytorch up quickly enough, but kinda don’t want to!