我在Colab中使用GPU和不使用GPU进行神经网络的定制训练,使用CPU的训练过程更快,这让我认为我没有并行操作或遗漏了什么。我不认为这是因为模型很小,因为我尝试了更复杂的模型,但问题仍然存在: ## Import libraries
import matplotlib
# matplotlib.use('TkAgg') # Required to make it run on both Windows and Mac
import matplotlib.pyplot as plt
import tensorflow as tf
from tensorflow impor