From cb8a8c7677f026ead748f0be24ee32745e91d2db Mon Sep 17 00:00:00 2001
From: Saifeddine ALOUI <aloui.seifeddine@gmail.com>
Date: Sat, 18 Nov 2023 02:45:54 +0100
Subject: [PATCH] fixed cuda install

---
 lollms/utilities.py | 6 +++++-
 1 file changed, 5 insertions(+), 1 deletion(-)

diff --git a/lollms/utilities.py b/lollms/utilities.py
index a781c4d..da6b738 100644
--- a/lollms/utilities.py
+++ b/lollms/utilities.py
@@ -56,8 +56,12 @@ def remove_text_from_string(string, text_to_find):
 
 
 # Pytorch and cuda tools
-def check_torch_version(min_version):
+def check_torch_version(min_version, min_cuda_versio=12):
     import torch
+
+    if "+" in torch.__version__ and int(torch.__version__.split("+")[-1][2:4])<min_cuda_versio:
+        return False
+
     # Extract torch version from __version__ attribute with regular expression
     current_version_float = float('.'.join(torch.__version__.split(".")[:2]))
     # Check if the current version meets or exceeds the minimum required version