ExternalVendorCode/lollms-webui
1
0
Fork 0
mirror of https://github.com/ParisNeo/lollms-webui.git synced 2025-01-29 15:44:12 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

Update app.py

Browse Source
This commit is contained in:
Saifeddine ALOUI 2023-11-15 14:07:10 +01:00 committed by GitHub
parent ab37ea40a7
commit 14e352848a
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 3 additions and 17 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

20
app.py
View File

@ -27,7 +27,7 @@ import traceback
import webbrowser
from pathlib import Path
from lollms.utilities import AdvancedGarbageCollector
from lollms.utilities import reinstall_pytorch_with_cuda
def run_update_script(args=None):
update_script = Path(__file__).parent/"update_script.py"
@ -1019,28 +1019,14 @@ try:
def upgrade_to_gpu(self):
ASCIIColors.yellow("Received command to upgrade to GPU")
ASCIIColors.info("Installing cuda toolkit")
res = subprocess.check_call(["conda", "install", "-c", "nvidia/label/cuda-11.7.0", "-c", "nvidia", "-c", "conda-forge", "cuda-toolkit", "ninja", "git", "--force-reinstall", "-y"])
if res!=0:
ASCIIColors.red("Couldn't install cuda toolkit")
return jsonify({'status':False, "error": "Couldn't install cuda toolkit. Make sure you are running from conda environment"})
ASCIIColors.green("Cuda toolkit installed successfully")
ASCIIColors.yellow("Removing pytorch")
try:
res = subprocess.check_call(["pip","uninstall","torch", "torchvision", "torchaudio", "-y"])
except :
pass
ASCIIColors.green("PyTorch unstalled successfully")
ASCIIColors.green("PyTorch uninstalled successfully")
reinstall_pytorch_with_cuda()
ASCIIColors.yellow("Installing pytorch with cuda support")
res = subprocess.check_call(["pip","install","--upgrade","torch==2.0.1+cu117", "torchvision", "torchaudio", "--index-url", "https://download.pytorch.org/whl/cu117","--no-cache"])
if res==0:
ASCIIColors.green("PyTorch installed successfully")
import torch
if torch.cuda.is_available():
ASCIIColors.success("CUDA is supported.")
else:
ASCIIColors.warning("CUDA is not supported. This may mean that the upgrade didn't succeed. Try rebooting the application")
else:
ASCIIColors.green("An error hapened")
self.config.enable_gpu=True
return jsonify({'status':res==0})