我最近更新了我的GPU,并转移到tf-nightly-gpu。不幸的是,我的更改破坏了我的代码。我一直在阅读关于这些变化的文章,我已经将我的代码剥离到下面的几行,以努力专注于这个问题。我拉出了指南的确切部分,并担心答案就在我面前,而我错过了它:https://www.tensorflow.org/api_docs/python/tf/keras/Sequential?version=nightly 下面是我的代码: import tensorflow as tf
# Define the
CUDA和CUDA DNN是按照Nvidia文档和测试正常安装的(关于CUDA文件夹中的示例)。Anaconda (conda 4.9.2), virtual environment created with python 3.8, installation of tensorflow from pip tf-nightly-gpu(tf-nightly-gpu==2.5.0.dev20201213).
我试图训练CNN对图像进行分类,我有一个Keras脚本,它成功地运行在CPU和GPU环境中,但是网络只在CPU环境中学习。在GPU环境中,损耗在第一个时期后减小,但在此之后保持不变。为什么?我尝试过不同的架构和不同的优化器,但是不管我对GPU环境有什么问题。import tensorflow as tf
from tensorflow.keras.layers import Inpu