disable cuda for classify.py

2024-12-26 16:30:19 +05:30
parent 1c71419537
commit d0f6c8c77a
1 changed files with 5 additions and 0 deletions
@@ -3,6 +3,11 @@ import tensorflow_hub as hub
 import librosa
 import numpy as np
 import csv
+import os
+
+# Disable CUDA
+os.environ["CUDA_VISIBLE_DEVICES"] = "-1"
+

 model = hub.load('https://tfhub.dev/google/yamnet/1')