AssertionError: Torch not compiled with CUDA enabled

此錯誤是由於下載的torch沒有cuda,在運行時就會出錯，經過查閱，在程序最開始的地方加上(在開始添加了各種包后面即可)：

device = torch.device("cuda" if torch.cuda.is_available() else "cpu")

代碼其余地方出現.cuda()的地方改成.to(device)就可以在無gpu的環境中運行了。