import torch print(torch.version.cuda) !pip install boto3 !pip install requests !pip install tqdm !pip install regex
查看gpu显存占用情况
单卡2080ti设定32显存会爆,设置16可以正常跑