I have tensorflow version=2.3.0 and cuda toolkit=10.1 and cudnn=7.6.5 all are compatiblw with each other but after then also code is running on cpu, i will give you the whole documenatation below please provide some solution
…
** CUDA_VISIBLE_DEVICES=5 /home/akashnegi/retina/retinanet/keras-retinanet-main/keras_retinanet/bin/train.py --gpu 5 csv /home/akashnegi/retina/retinanet/Retinanet-Tutorial/xray/annotation/train_label.csv /home/akashnegi/retina/retinanet/Retinanet-Tutorial/xray/class/class.csv
2024-01-19 06:14:53.761719: I tensorflow/stream_executor/platform/default/dso_loader.cc:48] Successfully opened dynamic library libcudart.so.10.1
Using TensorFlow backend.
2024-01-19 06:14:54.676816: I tensorflow/stream_executor/platform/default/dso_loader.cc:48] Successfully opened dynamic library libcuda.so.1
2024-01-19 06:14:54.744977: I tensorflow/core/common_runtime/gpu/gpu_device.cc:1716] Found device 0 with properties:
pciBusID: 0000:0b:00.0 name: Tesla V100-SXM2-32GB computeCapability: 7.0
coreClock: 1.53GHz coreCount: 80 deviceMemorySize: 31.75GiB deviceMemoryBandwidth: 836.37GiB/s
2024-01-19 06:14:54.745026: I tensorflow/stream_executor/platform/default/dso_loader.cc:48] Successfully opened dynamic library libcudart.so.10.1
2024-01-19 06:14:54.746702: I tensorflow/stream_executor/platform/default/dso_loader.cc:48] Successfully opened dynamic library libcublas.so.10
2024-01-19 06:14:54.748573: I tensorflow/stream_executor/platform/default/dso_loader.cc:48] Successfully opened dynamic library libcufft.so.10
2024-01-19 06:14:54.748863: I tensorflow/stream_executor/platform/default/dso_loader.cc:48] Successfully opened dynamic library libcurand.so.10
2024-01-19 06:14:54.750684: I tensorflow/stream_executor/platform/default/dso_loader.cc:48] Successfully opened dynamic library libcusolver.so.10
2024-01-19 06:14:54.751682: I tensorflow/stream_executor/platform/default/dso_loader.cc:48] Successfully opened dynamic library libcusparse.so.10
2024-01-19 06:14:54.755340: I tensorflow/stream_executor/platform/default/dso_loader.cc:48] Successfully opened dynamic library libcudnn.so.7
2024-01-19 06:14:54.757429: I tensorflow/core/common_runtime/gpu/gpu_device.cc:1858] Adding visible gpu devices: 0
Creating model, this may take a second…
2024-01-19 06:14:55.277390: 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.
2024-01-19 06:14:55.287450: I tensorflow/core/platform/profile_utils/cpu_utils.cc:104] CPU Frequency: 2194975000 Hz
2024-01-19 06:14:55.290323: I tensorflow/compiler/xla/service/service.cc:168] XLA service 0x55e9a303f910 initialized for platform Host (this does not guarantee that XLA will be used). Devices:
2024-01-19 06:14:55.290382: I tensorflow/compiler/xla/service/service.cc:176] StreamExecutor device (0): Host, Default Version
2024-01-19 06:14:55.291734: I tensorflow/core/common_runtime/gpu/gpu_device.cc:1257] Device interconnect StreamExecutor with strength 1 edge matrix:
2024-01-19 06:14:55.291749: I tensorflow/core/common_runtime/gpu/gpu_device.cc:1263]
Model: “retinanet”
**
Hi @Akash_Negi
Welcome to the TensorFlow Forum!
Please provide us some more details on the issue like which System OS you are using, the installed Python version and what steps you have followed to install the TensorFlow with GPU in your system. What code did you execute which shows the above error to understand the issue better?
Please refer to the official TensorFlow Install doc and verify if you have followed all the steps correctly to install TensorFlow with GPU support.
Let us know if the issue still persists. Thank you.