RuntimeError: CUDA out of memory. Tried to allocate 100.00 MiB (GPU 0; 10.76 GiB total capacity; 9.4 训练时模型的epoch和batch_size设置的太大超出了机器本身的配置; epoch和batch_size都相应的调整降低,比较极端的操作是全部都调整为1。