■事象
上記のコードをWindows10のJupyter Notebookで素直に実行するとエラーになりました。
以下、要修正点です。
◆要修正点
1.fake_image_batch = Tensor(4, 28 * 28)
正:fake_image_batch = Tensor(4, 28 * 28).cuda()
※「cuda()」を付けないと、fake_image_batchがCPU上で、モデルがGPU上なのでエラーになります。
2.tensorboard --logdir .
正:下記です。
%load_ext tensorboard
%tensorboard --logdir ./lightning_logs