编号55

tensorflow将代码从cpu运行改为gpu


提问于2019-08-10 10:04    231 已解决
tensorflow将代码从cpu运行改为gpu运行时是不是只用把loss和求梯度的那一部分放在gpu上? 还有神经网络里边的变量tf.Variable是否还要修改为tf.get_variable()

站长回答


抱歉,水平有限,了解不多,参见谷歌