1 pip安装相关软件
import torch print(torch.version.cuda) !pip install boto3 !pip install requests !pip install tqdm !pip install regex
查看gpu显存占用情况
2 在.ipynb中运行.py文件
3 注意batch size的设定
单卡2080ti设定32显存会爆,设置16可以正常跑
import torch print(torch.version.cuda) !pip install boto3 !pip install requests !pip install tqdm !pip install regex
查看gpu显存占用情况
单卡2080ti设定32显存会爆,设置16可以正常跑