mmarch/text-generation-webui
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity

Manually install CUDA runtime libraries

Browse source
This commit is contained in:
oobabooga 2023-10-12 21:02:44 -07:00
parent faf5c4dd58
commit 43be1be598
2 changed files with 10 additions and 5 deletions
Download patch file Download diff file Expand all files Collapse all files

4
one_click.py
View file

@ -187,6 +187,10 @@ def install_webui():
# Install Git and then Pytorch
run_cmd(f"{install_git} && {install_pytorch} && python -m pip install py-cpuinfo==9.0.0", assert_success=True, environment=True)
# Install CUDA libraries (this wasn't necessary for Pytorch before...)
if choice == "A":
run_cmd("conda install -y -c \"nvidia/label/cuda-11.8.0\" cuda-runtime", assert_success=True, environment=True)
# Install the webui requirements
update_requirements(initial_installation=True)