指定GPU

可以快速上手的开发文档

#指定GPU

在pytorch中为Module和Tensor指定GPU的例子

pytorch指定GPU在用pytorch写CNN的时候,发现一运行程序就卡住,然后cpu占用率100%,nvidia-smi查..