1.电脑有两个GPU 0和1，如何改代码使它使用GPU1呀。2.CUDA out of memory这个问题是因为什么呀

[appreciatehxy]

第二个问题：RuntimeError: CUDA out of memory. Tried to allocate 20.00 MiB (GPU 0; 2.00 GiB total capacity; 2.06 GiB already allocated; 0 bytes free; 2.29 GiB reserved in total by PyTorch)
batch_size我已经改成了4，但还是报错唉

[Fafa-DL]

https://zhuanlan.zhihu.com/p/166161217