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可以正常跑

 

posted @ 2020-11-09 00:28  冰河入梦~  阅读(29)  评论(0编辑  收藏  举报