Pytorch神经网络如何将数据集加载到GPU中



当将数据集加载到GPU中进行训练时,Pytorch NN会加载整个数据集还是只加载批次?

我有一个33GB的数据集,可以舒适地放在我的普通RAM(64GB(上,但我只有16GB的GPU RAM(T4(。只要Pytorch一次只将一个批加载到GPU中,就应该可以正常工作而不会出现任何内存问题?

您可以一次将一批数据加载到GPU中。您应该使用数据加载器来获取一批数据,并初始化一个torch设备实例以使用GPU。

您可以查看以下教程。它使用数据加载器来批量获取数据,并使用torch设备将数据加载到GPU。

https://pytorch.org/tutorials/beginner/transfer_learning_tutorial.html