@kjackson for me it worked. Exactly how you described it. I checked out the git and run the notebook as it is without any changes from top to bottom without errors. I used the conda_pytorch_p36 kernel.