check cuda and cudnn

app.py

@@ -1,4 +1,5 @@
 import spaces
+import tensorflow as tf
 
 import gradio as gr
 import os
@@ -22,6 +23,10 @@ class App:
         else:
             print("Use Open AI Whisper implementation")
         print(f"Device \"{self.whisper_inf.device}\" is detected")
+        cuda_version = torch.version.cuda
+        cudnn_version = tf.sysconfig.get_build_info()["cuda"]["cudnn_version"]
+        print(f"cuDNN version: {cudnn_version}")
+        print(f"CUDA version: {cuda_version}")
         self.nllb_inf = NLLBInference()
         self.deepl_api = DeepLAPI()
 

packages.txt

@@ -1 +0,0 @@
-nvidia-cuda-toolkit