This does not help either. In case you want to reproduce:
Python 3.8.2
torch==1.4.0
$ jupyter --version
jupyter core     : 4.6.3
jupyter-notebook : 6.0.3
qtconsole        : 4.7.2
ipython          : 7.13.0
ipykernel        : 5.2.0
jupyter client   : 6.1.2
jupyter lab      : 2.0.1
nbconvert        : 5.6.1
ipywidgets       : 7.5.1
nbformat         : 5.0.5
traitlets        : 4.3.3

I have tired also some other stuff and did not get it working.