Issue Installing cuDNN via setup-3.10.bat in Anaconda Environment

[dexterhuh]

Context:

I am using Anaconda for Python environment management on Windows 10, and Kohya is running within an Anaconda Python environment. The initial setup includes PyTorch 2.1.2+cu118+cuDNN 8700, and after running the setup-3.10.bat script, the Kohya GUI works fine. But I cuDNN cannot be installed successfully.

Steps to Reproduce:

1. Run setup-3.10.bat.
2. Select option 2 (Optional) to install cuDNN files (to use the latest supported cuDNN version).

Actual Result:

The installation fails with the following error:

16:02:15-012406 INFO Installing nvidia-cudnn-cu11 8.9.5.29...
16:02:15-014407 INFO Installing package: nvidia-cudnn-cu11==8.9.5.29
16:02:49-088459 ERROR Error running pip: install --upgrade --upgrade nvidia-cudnn-cu11==8.9.5.29
16:02:49-091460 INFO Copying CUDNN files from E:\Anaconda\KohyaSS\kohya_ss\venv\Lib\site-packages\nvidia\cudnn\bin
to E:\Anaconda\KohyaSS\Lib\site-packages\torch\lib...
16:02:49-093461 ERROR Installation Failed: "E:\Anaconda\KohyaSS\kohya_ss\venv\Lib\site-packages\nvidia\cudnn\bin"
could not be found.

Expected Result:

The script should successfully install the latest version of cuDNN (cuDNN 8.9.5.29).

My Guess:

Since I am using an Anaconda environment, running the setup-3.10.bat script may not create the venv\Lib\site-packages folder. Instead, the site-packages folder is located within the Anaconda environment itself. Does the cuDNN installation script need to be adapted to support multiple Python environments when using setup-3.10.bat?

Manual Solution:

I was able to resolve the issue by manually creating the directory venv\Lib\site-packages\nvidia\cudnn, and then I manually downloaded cuDNN 8.9.6.50 (cudnn-windows-x86_64-8.9.6.50_cuda11-archive.zip) and extracted it into that directory. Afterward, I ran the setup-3.10.bat script again and selected 2 (Optional) to install cuDNN files, and the installation was successful.