ExternalVendorCode/lollms
1
0
Fork 0
mirror of https://github.com/ParisNeo/lollms.git synced 2025-04-05 09:59:08 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

Update utilities.py

Browse Source
This commit is contained in:
Saifeddine ALOUI 2024-03-04 11:29:15 +01:00 committed by GitHub
parent de39e5a86d
commit ab5a195b40
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 8 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

9
lollms/utilities.py
View File

@ -446,7 +446,14 @@ def install_cuda():
result = conda.cli.main("install", "-c", "nvidia/label/cuda-12.3.2", "-c", "nvidia", "-c", "conda-forge", "cuda-compiler", "-y","--force-reinstall")
except Exception as ex:
ASCIIColors.error(ex)
def install_cmake():
import conda.cli
try:
ASCIIColors.info("Installing cmake") # -c nvidia/label/cuda-12.3.2 -c nvidia -c conda-forge
result = conda.cli.main("install", "-c", "nvidia/label/cuda-12.3.2", "-c", "nvidia", "-c", "conda-forge", "cmake","-y","--force-reinstall")
except Exception as ex:
ASCIIColors.error(ex)
def reinstall_pytorch_with_cuda():
try: