【AI】PyTorch Lightning LIGHTNING IN 15 MINUTESエラー | 夕湖津のブログ

夕湖津のブログ

問題解決に役立つ情報の提供を目指します

■事象

 

 

上記のコードを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