import error when executing cifar10.py

[ace510]

expected behavior: script completes

observed behavior: script fails with following error

python cifar10.py 2024-08-13 14:56:40.508016: I external/local_xla/xla/tsl/cuda/cudart_stub.cc:32] Could not find cuda drivers on your machine, GPU will not be used. 2024-08-13 14:56:40.512896: I external/local_xla/xla/tsl/cuda/cudart_stub.cc:32] Could not find cuda drivers on your machine, GPU will not be used. 2024-08-13 14:56:40.531670: E external/local_xla/xla/stream_executor/cuda/cuda_fft.cc:485] Unable to register cuFFT factory: Attempting to register factory for plugin cuFFT when one has already been registered 2024-08-13 14:56:40.562910: E external/local_xla/xla/stream_executor/cuda/cuda_dnn.cc:8454] Unable to register cuDNN factory: Attempting to register factory for plugin cuDNN when one has already been registered 2024-08-13 14:56:40.572102: E external/local_xla/xla/stream_executor/cuda/cuda_blas.cc:1452] Unable to register cuBLAS factory: Attempting to register factory for plugin cuBLAS when one has already been registered 2024-08-13 14:56:42.602908: W tensorflow/compiler/tf2tensorrt/utils/py_utils.cc:38] TF-TRT Warning: Could not find TensorRT Traceback (most recent call last): File "/home/cetadmin/Github/cnn-without-any-downsampling/cifar10.py", line 8, in <module> from tensorflow_model_optimization.python.core.sparsity.keras import prune ModuleNotFoundError: No module named 'tensorflow_model_optimization'

python --version Python 3.12.3