环境 tf2.5 kears2.5.0
cuda 11.2 cudnn8.2
然后会出现这种问题
D:\anaconda3\envs\TF2.1\python.exe D:/1dcnn/keras_bearing_fault_diagnosis-master/keras_bearing_fault_diagnosis-master/cnn_1D.py
2022-05-09 14:18:18.314339: I tensorflow/stream_executor/platform/default/dso_loader.cc:53] Successfully opened dynamic library cudart64_110.dll
训练样本维度: (2800, 2048, 1)
2800 训练样本个数
验证样本的维度 (800, 2048, 1)
800 验证样本个数
测试样本的维度 (400, 2048, 1)
400 测试样本个数
2022-05-09 14:18:20.916216: I tensorflow/stream_executor/platform/default/dso_loader.cc:53] Successfully opened dynamic library nvcuda.dll
2022-05-09 14:18:21.951580: I tensorflow/core/common_runtime/gpu/gpu_device.cc:1733] Found device 0 with properties:
pciBusID: 0000:01:00.0 name: NVIDIA GeForce GTX 1650 computeCapability: 7.5
coreClock: 1.56GHz coreCount: 16 deviceMemorySize: 4.00GiB deviceMemoryBandwidth: 119.24GiB/s
2022-05-09 14:18:21.951820: I tensorflow/stream_executor/platform/default/dso_loader.cc:53] Successfully opened dynamic library cudart64_110.dll
2022-05-09 14:18:21.957774: I tensorflow/stream_executor/platform/default/dso_loader.cc:53] Successfully opened dynamic library cublas64_11.dll
2022-05-09 14:18:21.957875: I tensorflow/stream_executor/platform/default/dso_loader.cc:53] Successfully opened dynamic library cublasLt64_11.dll
2022-05-09 14:18:22.325343: I tensorflow/stream_executor/platform/default/dso_loader.cc:53] Successfully opened dynamic library cufft64_10.dll
2022-05-09 14:18:22.372285: I tensorflow/stream_executor/platform/default/dso_loader.cc:53] Successfully opened dynamic library curand64_10.dll
2022-05-09 14:18:22.379496: I tensorflow/stream_executor/platform/default/dso_loader.cc:53] Successfully opened dynamic library cusolver64_11.dll
2022-05-09 14:18:22.381823: I tensorflow/stream_executor/platform/default/dso_loader.cc:53] Successfully opened dynamic library cusparse64_11.dll
2022-05-09 14:18:22.382256: I tensorflow/stream_executor/platform/default/dso_loader.cc:53] Successfully opened dynamic library cudnn64_8.dll
2022-05-09 14:18:22.382379: I tensorflow/core/common_runtime/gpu/gpu_device.cc:1871] Adding visible gpu devices: 0
2022-05-09 14:18:22.382663: I tensorflow/core/platform/cpu_feature_guard.cc:142] This TensorFlow binary is optimized with oneAPI Deep Neural Network Library (oneDNN) to use the following CPU instructions in performance-critical operations: AVX AVX2
To enable them in other operations, rebuild TensorFlow with the appropriate compiler flags.
2022-05-09 14:18:22.383279: I tensorflow/core/common_runtime/gpu/gpu_device.cc:1733] Found device 0 with properties:
pciBusID: 0000:01:00.0 name: NVIDIA GeForce GTX 1650 computeCapability: 7.5
coreClock: 1.56GHz coreCount: 16 deviceMemorySize: 4.00GiB deviceMemoryBandwidth: 119.24GiB/s
2022-05-09 14:18:22.383542: I tensorflow/core/common_runtime/gpu/gpu_device.cc:1871] Adding visible gpu devices: 0
2022-05-09 14:18:22.870153: I tensorflow/core/common_runtime/gpu/gpu_device.cc:1258] Device interconnect StreamExecutor with strength 1 edge matrix:
2022-05-09 14:18:22.870253: I tensorflow/core/common_runtime/gpu/gpu_device.cc:1264] 0
2022-05-09 14:18:22.870307: I tensorflow/core/common_runtime/gpu/gpu_device.cc:1277] 0: N
2022-05-09 14:18:22.870488: I tensorflow/core/common_runtime/gpu/gpu_device.cc:1418] Created TensorFlow device (/job:localhost/replica:0/task:0/device:GPU:0 with 2143 MB memory) -> physical GPU (device: 0, name: NVIDIA GeForce GTX 1650, pci bus id: 0000:01:00.0, compute capability: 7.5)
2022-05-09 14:18:22.982294: I tensorflow/core/profiler/lib/profiler_session.cc:126] Profiler session initializing.
2022-05-09 14:18:22.982369: I tensorflow/core/profiler/lib/profiler_session.cc:141] Profiler session started.
2022-05-09 14:18:22.982446: I tensorflow/core/profiler/internal/gpu/cupti_tracer.cc:1611] Profiler found 1 GPUs
2022-05-09 14:18:22.983221: W tensorflow/stream_executor/platform/default/dso_loader.cc:64] Could not load dynamic library 'cupti64_112.dll'; dlerror: cupti64_112.dll not found
2022-05-09 14:18:22.983893: W tensorflow/stream_executor/platform/default/dso_loader.cc:64] Could not load dynamic library 'cupti.dll'; dlerror: cupti.dll not found
2022-05-09 14:18:22.984000: E tensorflow/core/profiler/internal/gpu/cupti_tracer.cc:1661] function cupti_interface_->Subscribe( &subscriber_, (CUpti_CallbackFunc)ApiCallback, this)failed with error CUPTI could not be loaded or symbol could not be found.
2022-05-09 14:18:22.984153: I tensorflow/core/profiler/lib/profiler_session.cc:159] Profiler session tear down.
2022-05-09 14:18:22.984226: E tensorflow/core/profiler/internal/gpu/cupti_tracer.cc:1752] function cupti_interface_->Finalize()failed with error CUPTI could not be loaded or symbol could not be found.
2022-05-09 14:18:23.184377: I tensorflow/compiler/mlir/mlir_graph_optimization_pass.cc:176] None of the MLIR Optimization Passes are enabled (registered 2)
Epoch 1/12
Traceback (most recent call last):
File "D:\1dcnn\keras_bearing_fault_diagnosis-master\keras_bearing_fault_diagnosis-master\cnn_1D.py", line 77, in <module>
model.fit(x=x_train, y=y_train, batch_size=batch_size, epochs=epochs,
File "D:\anaconda3\envs\TF2.1\lib\site-packages\keras\engine\training.py", line 1158, in fit
tmp_logs = self.train_function(iterator)
File "D:\anaconda3\envs\TF2.1\lib\site-packages\tensorflow\python\eager\def_function.py", line 889, in __call__
result = self._call(*args, **kwds)
File "D:\anaconda3\envs\TF2.1\lib\site-packages\tensorflow\python\eager\def_function.py", line 933, in _call
self._initialize(args, kwds, add_initializers_to=initializers)
File "D:\anaconda3\envs\TF2.1\lib\site-packages\tensorflow\python\eager\def_function.py", line 763, in _initialize
self._stateful_fn._get_concrete_function_internal_garbage_collected( # pylint: disable=protected-access
File "D:\anaconda3\envs\TF2.1\lib\site-packages\tensorflow\python\eager\function.py", line 3050, in _get_concrete_function_internal_garbage_collected
graph_function, _ = self._maybe_define_function(args, kwargs)
File "D:\anaconda3\envs\TF2.1\lib\site-packages\tensorflow\python\eager\function.py", line 3444, in _maybe_define_function
graph_function = self._create_graph_function(args, kwargs)
File "D:\anaconda3\envs\TF2.1\lib\site-packages\tensorflow\python\eager\function.py", line 3279, in _create_graph_function
func_graph_module.func_graph_from_py_func(
File "D:\anaconda3\envs\TF2.1\lib\site-packages\tensorflow\python\framework\func_graph.py", line 999, in func_graph_from_py_func
func_outputs = python_func(*func_args, **func_kwargs)
File "D:\anaconda3\envs\TF2.1\lib\site-packages\tensorflow\python\eager\def_function.py", line 672, in wrapped_fn
out = weak_wrapped_fn().__wrapped__(*args, **kwds)
File "D:\anaconda3\envs\TF2.1\lib\site-packages\tensorflow\python\framework\func_graph.py", line 986, in wrapper
raise e.ag_error_metadata.to_exception(e)
ValueError: in user code:
D:\anaconda3\envs\TF2.1\lib\site-packages\keras\engine\training.py:830 train_function *
return step_function(self, iterator)
D:\anaconda3\envs\TF2.1\lib\site-packages\keras\engine\training.py:813 run_step *
outputs = model.train_step(data)
D:\anaconda3\envs\TF2.1\lib\site-packages\keras\engine\training.py:771 train_step *
loss = self.compiled_loss(
D:\anaconda3\envs\TF2.1\lib\site-packages\keras\engine\compile_utils.py:201 __call__ *
loss_value = loss_obj(y_t, y_p, sample_weight=sw)
D:\anaconda3\envs\TF2.1\lib\site-packages\keras\losses.py:142 __call__ *
losses = call_fn(y_true, y_pred)
D:\anaconda3\envs\TF2.1\lib\site-packages\keras\losses.py:246 call *
return ag_fn(y_true, y_pred, **self._fn_kwargs)
D:\anaconda3\envs\TF2.1\lib\site-packages\tensorflow\python\util\dispatch.py:206 wrapper **
return target(*args, **kwargs)
D:\anaconda3\envs\TF2.1\lib\site-packages\keras\losses.py:1630 categorical_crossentropy
return backend.categorical_crossentropy(
D:\anaconda3\envs\TF2.1\lib\site-packages\tensorflow\python\util\dispatch.py:206 wrapper
return target(*args, **kwargs)
D:\anaconda3\envs\TF2.1\lib\site-packages\keras\backend.py:4827 categorical_crossentropy
target.shape.assert_is_compatible_with(output.shape)
D:\anaconda3\envs\TF2.1\lib\site-packages\tensorflow\python\framework\tensor_shape.py:1161 assert_is_compatible_with
raise ValueError("Shapes %s and %s are incompatible" % (self, other))
ValueError: Shapes (None, 4) and (None, 10) are incompatible
Process finished with exit code 1
>