【平台使用】如何使用Pytorch

请问如何在代码中使用自定义的Pytorch神经网络并使用GPU加速？目前在文档中没有找到相关的描述，是否支持这种功能呢？