ExternalVendorCode/lollms
1
0
Fork 0
mirror of https://github.com/ParisNeo/lollms.git synced 2025-04-16 06:56:33 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

fixed cuda install

Browse Source
This commit is contained in:
Saifeddine ALOUI 2023-11-18 02:45:54 +01:00
parent 0de9bc8713
commit cb8a8c7677
1 changed files with 5 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
lollms/utilities.py
View File

@ -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