mirror of
https://github.com/opendatalab/MinerU.git
synced 2026-03-30 20:48:38 +07:00
11 lines
152 B
Python
11 lines
152 B
Python
import pytest
|
|
import torch
|
|
|
|
def clear_gpu_memory():
|
|
'''
|
|
clear GPU memory
|
|
'''
|
|
torch.cuda.empty_cache()
|
|
print("GPU memory cleared.")
|
|
|