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

preparing

Browse Source
This commit is contained in:
Saifeddine ALOUI 2023-12-31 16:41:20 +01:00
parent 1a85b58c24
commit df9022b582
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
lollms/utilities.py
View File

@ -180,7 +180,7 @@ def check_torch_version(min_version, min_cuda_versio=12):
def reinstall_pytorch_with_cuda():
try:
ASCIIColors.info("Installing cuda 12.1.1")
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", "cuda-toolkit", "-y"])
except Exception as ex:
ASCIIColors.error(ex)