我正在使用 google colab pro。 我在 GPU 模式下运行:
device = torch.device('cuda' if torch.cuda.is_available() else 'cpu')
print(device)
print(torch.cuda.get_device_name())
返回:
cuda
Tesla P100-PCIE-16GB
我能够创建一个模型并完美地运行它,然后保存它。
当我尝试在没有 to(device)
的情况下加载它时,它可以工作。
但是当我跑步时:
notes_model = LSTM_notes(em_dim).to(device)
会话崩溃:
Your session crashed for an unknown reason. View runtime logs
我曾尝试查看日志,但它们似乎没有提供有意义的信息:
May 20, 2021, 2:37:08 PM WARNING WARNING:root:kernel 53812acc-3a19-48e2-a41b-1acaab9dfe61 restarted
May 20, 2021, 2:37:08 PM INFO KernelRestarter: restarting kernel (1/5), keep random ports
May 20, 2021, 2:36:44 PM WARNING WARNING:root:kernel 53812acc-3a19-48e2-a41b-1acaab9dfe61 restarted
May 20, 2021, 2:36:44 PM INFO KernelRestarter: restarting kernel (1/5), keep random ports
关于为什么会发生这种情况以及我可以做些什么来解决它的任何想法?
我认为这不是内存不足的问题。