Install nvidia-cudnn11==8.6.0.163 as workaround to NHCW/NCHW issue, ht @bertsky

This commit is contained in:
kba 2025-09-29 19:18:16 +02:00
parent a6f0af07d1
commit 0d695a2e2d
2 changed files with 3 additions and 5 deletions

View file

@ -77,6 +77,7 @@ build:
# Install with pip
install:
$(PIP) install .$(and $(EXTRAS),[$(EXTRAS)])
pip install nvidia-cudnn-cu11==8.6.0.163
# Install editable with pip
install-dev: