Getting the nvidia containers to work on a 3090

Hey gang. I’ve tried everything to get tensorflow to work on my 3090, so far no dice.
The latest thing I tried was actually using the Nvidia supplied containers. It appears the provided container uses CUDA 10 instead of 11, and seems to be hanging when I try to use a keras layer. Has anyone made this work?