From c74ab33905ab9d03bac4389bb802ff5f8976acb3 Mon Sep 17 00:00:00 2001
From: romi1502 <rhennequin@deezer.com>
Date: Fri, 25 Sep 2020 13:15:19 +0200
Subject: [PATCH] Changed gpu availability function to avoid deprectation
 warning

---
 spleeter/separator.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/spleeter/separator.py b/spleeter/separator.py
index 12f01b7..47305be 100644
--- a/spleeter/separator.py
+++ b/spleeter/separator.py
@@ -69,7 +69,7 @@ def get_backend(backend):
     # print("USING TENSORFLOW BACKEND !!!!!!")
     # return "tensorflow"
     if backend == "auto":
-        return "tensorflow" if tf.test.is_gpu_available() else "librosa"
+        return "tensorflow" if len(tf.config.list_physical_devices('GPU')) else "librosa"
     return backend