From de507627a718a53178e16dc388904e49fb0290d9 Mon Sep 17 00:00:00 2001 From: Saifeddine ALOUI Date: Wed, 31 Jan 2024 21:59:17 +0100 Subject: [PATCH] fixed --- lollms/utilities.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/lollms/utilities.py b/lollms/utilities.py index 6cdcb77..38b5f4f 100644 --- a/lollms/utilities.py +++ b/lollms/utilities.py @@ -340,17 +340,17 @@ def reinstall_pytorch_with_cuda(): try: import conda.cli ASCIIColors.info("Installing cuda 12.1.1") # -c nvidia/label/cuda-12.1.1 -c nvidia -c conda-forge - result = conda.cli.main(["conda", "install", "-c", "nvidia/label/cuda-12.1.1", "-c", "nvidia", "-c", "conda-forge", "cuda-toolkit","-y"]) + result = conda.cli.main("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 = conda.cli.main(["conda", "install", "-c", "nvidia/label/cuda-12.1.1", "-c", "nvidia", "-c", "conda-forge", "ninja", "-y"]) + result = conda.cli.main("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 = conda.cli.main(["conda", "install", "-c", "nvidia/label/cuda-12.1.1", "-c", "nvidia", "-c", "conda-forge", "cuda-compiler", "-y"]) + result = conda.cli.main("conda", "install", "-c", "nvidia/label/cuda-12.1.1", "-c", "nvidia", "-c", "conda-forge", "cuda-compiler", "-y") except Exception as ex: ASCIIColors.error(ex) try: