1024lab-How to run project using .ipynb
1 pip安装相关软件
import torch print(torch.version.cuda) !pip install boto3 !pip install requests !pip install tqdm !pip install regex
!pip install sklearn
查看gpu显存占用情况
2 在.ipynb中运行.py文件
3 注意batch size的设定
单卡2080ti设定32显存会爆,设置16可以正常跑
本文来自博客园,作者:冰河入梦~,转载请注明原文链接:https://www.cnblogs.com/xuechengmeigui/p/13946624.html