I cann't slove this issue while running train_triplet.py

[ayushkoirala]

2020-11-10 14:38:04.100728: W tensorflow/stream_executor/platform/default/dso_loader.cc:59] Could not load dynamic library 'libcudart.so.10.1'; dlerror: libcudart.so.10.1: cannot open shared object file: No such file or directory 2020-11-10 14:38:04.100756: I tensorflow/stream_executor/cuda/cudart_stub.cc:29] Ignore above cudart dlerror if you do not have a GPU set up on your machine. Saved model not found, loading untrained FaceNet 2020-11-10 14:38:06.415020: W tensorflow/stream_executor/platform/default/dso_loader.cc:59] Could not load dynamic library 'libcuda.so.1'; dlerror: libcuda.so.1: cannot open shared object file: No such file or directory 2020-11-10 14:38:06.415055: W tensorflow/stream_executor/cuda/cuda_driver.cc:312] failed call to cuInit: UNKNOWN ERROR (303) 2020-11-10 14:38:06.415090: I tensorflow/stream_executor/cuda/cuda_diagnostics.cc:156] kernel driver does not appear to be running on this host (ayush-Inspiron-5558): /proc/driver/nvidia/version does not exist 2020-11-10 14:38:06.415374: 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: AVX2 FMA To enable them in other operations, rebuild TensorFlow with the appropriate compiler flags. 2020-11-10 14:38:06.422818: I tensorflow/core/platform/profile_utils/cpu_utils.cc:104] CPU Frequency: 2394610000 Hz 2020-11-10 14:38:06.423240: I tensorflow/compiler/xla/service/service.cc:168] XLA service 0x4022760 initialized for platform Host (this does not guarantee that XLA will be used). Devices: 2020-11-10 14:38:06.423298: I tensorflow/compiler/xla/service/service.cc:176] StreamExecutor device (0): Host, Default Version 2020-11-10 14:39:21.909641: I tensorflow/core/profiler/lib/profiler_session.cc:164] Profiler session started. WARNING:tensorflow:From train_triplet.py:82: Model.fit_generator (from tensorflow.python.keras.engine.training) is deprecated and will be removed in a future version. Instructions for updating: Please use Model.fit, which supports generators. Traceback (most recent call last): File "train_triplet.py", line 82, in tripletModel.fit_generator(gen, epochs=NUM_EPOCHS,steps_per_epoch=STEPS_PER_EPOCH,callbacks=[early_stopping, tensorboard]) File "/home/ayush/Desktop/FaceNet-Real-Time-face-recognition-master/env/lib/python3.6/site-packages/tensorflow/python/util/deprecation.py", line 324, in new_func return func(*args, **kwargs) File "/home/ayush/Desktop/FaceNet-Real-Time-face-recognition-master/env/lib/python3.6/site-packages/tensorflow/python/keras/engine/training.py", line 1829, in fit_generator initial_epoch=initial_epoch) File "/home/ayush/Desktop/FaceNet-Real-Time-face-recognition-master/env/lib/python3.6/site-packages/tensorflow/python/keras/engine/training.py", line 108, in _method_wrapper return method(self, *args, **kwargs) File "/home/ayush/Desktop/FaceNet-Real-Time-face-recognition-master/env/lib/python3.6/site-packages/tensorflow/python/keras/engine/training.py", line 1098, in fit tmp_logs = train_function(iterator) File "/home/ayush/Desktop/FaceNet-Real-Time-face-recognition-master/env/lib/python3.6/site-packages/tensorflow/python/eager/def_function.py", line 780, in call result = self._call(*args, **kwds) File "/home/ayush/Desktop/FaceNet-Real-Time-face-recognition-master/env/lib/python3.6/site-packages/tensorflow/python/eager/def_function.py", line 840, in _call return self._stateless_fn(*args, **kwds) File "/home/ayush/Desktop/FaceNet-Real-Time-face-recognition-master/env/lib/python3.6/site-packages/tensorflow/python/eager/function.py", line 2829, in call return graph_function._filtered_call(args, kwargs) # pylint: disable=protected-access File "/home/ayush/Desktop/FaceNet-Real-Time-face-recognition-master/env/lib/python3.6/site-packages/tensorflow/python/eager/function.py", line 1848, in _filtered_call cancellation_manager=cancellation_manager) File "/home/ayush/Desktop/FaceNet-Real-Time-face-recognition-master/env/lib/python3.6/site-packages/tensorflow/python/eager/function.py", line 1924, in _call_flat ctx, args, cancellation_manager=cancellation_manager)) File "/home/ayush/Desktop/FaceNet-Real-Time-face-recognition-master/env/lib/python3.6/site-packages/tensorflow/python/eager/function.py", line 550, in call ctx=ctx) File "/home/ayush/Desktop/FaceNet-Real-Time-face-recognition-master/env/lib/python3.6/site-packages/tensorflow/python/eager/execute.py", line 60, in quick_execute inputs, attrs, num_outputs) tensorflow.python.framework.errors_impl.InvalidArgumentError: Default MaxPoolingOp only supports NHWC on device type CPU [[node functional_1/FaceRecoModel/max_pooling2d/MaxPool_2 (defined at train_triplet.py:82) ]] [Op:__inference_train_function_21540]

Function call stack: train_function