【RuntimeError: CUDA error: out of memory.
CUDA kernel errors might be asynchronously reported at some other API call,so the stacktrace below might be incorrect.
For debugging consider passing CUDA_LAUNCH_BLOCKING=1.】CUDA运行模式问题
引入CUDA_LAUNCH_BLOCKING=1
‘’‘程序文件内容开头加入’‘’
import os
os.environ['CUDA_LAUNCH_BLOCKING'] = '1'
用cpu运行
把.to('cuda')变成.to('cpu')就能获得正常的track