[WrtCdEvrydy]

You can actually add nightly-gpu-py3-jupyter if you'd like GPU level tensorflow as well :D

[Buetol]

Yup but it also needs the nvidia-docker image for the driver, but got it working in less than 10 minutes too !