requirement:tensorflow-cpu

[nibingwei123]

why the cuda and cudnn installed and the requirements file in docker install tensorflow-cpu version???

[Bhazantri]

modify the requirements.txt to explicitly install the GPU-compatible version of TensorFlow by specifying tensorflow-gpu or tensorflow==2.x+nv instead of tensorflow-cpu. Ensure your Dockerfile uses a GPU-enabled base image, such as nvidia/cuda:12.2.0-cudnn8-runtime-ubuntu20.04, and install TensorFlow accordingly. Additionally, use the --gpus all flag when running the container to enable GPU support, and verify that TensorFlow detects the GPU with tf.config.list_physical_devices('GPU').