显存不够的入门解决方法：

1. 利用`nvidia-smi`命令查看各个显卡的使用情况。
2. 用pytorch指定显卡。
`device = torch.device('cuda:'+str(1)) # 指定显卡1`