我正在使用 Ubuntu 16.04，并运行我的程序来训练深度学习模型。纪元数看起来很大，但程序太慢了。我想确保我的程序在 GPU 上运行？
我怎么知道呢？
谢谢！
cpu gpu deep-learning
cpu ×1
deep-learning ×1
gpu ×1