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

added remove pytorch

Browse Source
This commit is contained in:
Saifeddine ALOUI 2023-08-31 23:56:40 +02:00
parent 587b3f0451
commit 66abfac334
1 changed files with 4 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
app.py
View File

@ -855,8 +855,11 @@ class LoLLMsWebUI(LoLLMsAPPI):
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")
res = subprocess.check_call(["pip","uninstall","torch", "torchvision", "torchaudio"])
ASCIIColors.green("PyTorch unstalled successfully")
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"])
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