From de5e73c90f8b4bac5a636a8cb51a80902b2da4b4 Mon Sep 17 00:00:00 2001
From: Saifeddine ALOUI <aloui.seifeddine@gmail.com>
Date: Thu, 4 Jan 2024 04:40:11 +0100
Subject: [PATCH] ok
---
lollms/utilities.py | 8 ++++++++
1 file changed, 8 insertions(+)
diff --git a/lollms/utilities.py b/lollms/utilities.py
index ec2caad..e900b1f 100644
--- a/lollms/utilities.py
+++ b/lollms/utilities.py
@@ -182,7 +182,15 @@ def reinstall_pytorch_with_cuda():
try:
ASCIIColors.info("Installing cuda 12.1.1") # -c nvidia/label/cuda-12.1.1 -c nvidia -c conda-forge
result = subprocess.run(["conda", "install", "-c", "nvidia/label/cuda-12.1.1", "-c", "nvidia", "-c", "conda-forge", "cuda-toolkit","-y"])
+ except Exception as ex:
+ ASCIIColors.error(ex)
+ try:
+ ASCIIColors.info("Installing ninja") # -c nvidia/label/cuda-12.1.1 -c nvidia -c conda-forge
result = subprocess.run(["conda", "install", "-c", "nvidia/label/cuda-12.1.1", "-c", "nvidia", "-c", "conda-forge", "ninja", "-y"])
+ except Exception as ex:
+ ASCIIColors.error(ex)
+ try:
+ ASCIIColors.info("Installing cuda compiler") # -c nvidia/label/cuda-12.1.1 -c nvidia -c conda-forge
result = subprocess.run(["conda", "install", "-c", "nvidia/label/cuda-12.1.1", "-c", "nvidia", "-c", "conda-forge", "cuda-compiler", "-y"])
except Exception as ex:
ASCIIColors.error(ex)