【RuntimeError: CUDA out of memory. Tried to allocate 100.00 MiB (GPU 0; 10.76 GiB total capacity;】

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。

你可能感兴趣的:(日常问题,python)