完整错误:
RuntimeError: CUDA error: misaligned address
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.
其实就是显存不够。
并且后面的解决办法,在代码最开始处加:
import os
os.environ['CUDA_LAUNCH_BLOCKING'] = '1'
os.environ['CUDA_LAUNCH_BLOCKING'] = '1'含义:
把CUDA_LAUNCH_BLOCKING这个环境变量设置为1,1表示True,则会强制消除这种异步性。 具体解释参考:CUDA_LAUNCH_BLOCKING=1的作用-CSDN博客