PyTorch指定GPU的方法
Posted
tags:
篇首语：本文由小常识网(cha138.com)小编为大家整理，主要介绍了PyTorch指定GPU的方法相关的知识，希望对你有一定的参考价值。
参考技术A 在生成网络对象之前执行： torch.cuda.set_device(0)net = torch.nn.DataParallel(model, device_ids=[0])
net = Net.cuda(0)
以上是关于PyTorch指定GPU的方法的主要内容，如果未能解决你的问题，请参考以下文章