1580158
/
kohya_ss
mirror of https://github.com/bmaltais/kohya_ss
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

Move cudnn out of requirements

pull/2066/head
bmaltais 2024-03-11 16:50:51 -04:00
parent eb9d7bce05
commit 3be05253bb
2 changed files with 7 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
requirements_windows_torch2.txt
View File

@ -1,5 +1,4 @@
torch==2.1.2+cu118 torchvision==0.16.2+cu118 torchaudio==2.1.2+cu118 xformers==0.0.23.post1+cu118 --index-url https://download.pytorch.org/whl/cu118
nvidia-cudnn-cu11==8.9.5.29
bitsandbytes==0.43.0
tensorboard==2.14.1 tensorflow==2.14.0
-r requirements.txt

7
setup/setup_windows.py
View File

@ -15,6 +15,13 @@ RESET_COLOR = "\033[0m"
def cudnn_install():
log.info("Installing nvidia-cudnn-cu11 8.9.5.29...")
setup_common.install(
"--upgrade nvidia-cudnn-cu11==8.9.5.29",
"nvidia-cudnn-cu11 8.9.5.29",
reinstall=True,
)
# Original path with "..\\venv"
original_path = os.path.join(os.path.dirname(os.path.realpath(__file__)), "..\\venv\\Lib\\site-packages\\nvidia\\cudnn\\bin")
# Normalize the path to resolve "..\\venv"