Open araza008 opened 3 months ago
Using the command pip install --extra-index-url https://download.pytorch.org/whl/cu121 torch==2.2.0+cu121
on Linux (I know not Windows) I see as a dependency it installs nvidia-cudnn-cu12==8.9.2.26
. This looks like a direct dependency of torch
. Since we don't have a Windows build of cuDNN 8.9.2.26 available I'm not sure how this particular version of Torch would be installable on Windows.
I checked the METADATA
file and it shows a platform specific condition (Linux x86)
$ grep nvidia-cudnn torch-2.2.0+cu121.dist-info/METADATA
Requires-Dist: nvidia-cudnn-cu12 (==8.9.2.26) ; platform_system == "Linux" and platform_machine == "x86_64"
I'm not too familiar with Torch's dependencies, but I'm wondering if on Windows they are expecting you to install these libraries manually. Is Poetry maybe disregarding these platform hints?
I will need to repeat this on a Windows machine when I get an opportunity but wanted to provide some debug information to get started.
Hi @zeroepoch. Were you able to repro the issue on Windows platform? When is this expected to be resolved?
@araza008 the solution to this issue is to not use Poetry. PyTorch is essentially not compatible with Poetry on Windows from what I can see. Since it doesn't consider the platform conditions in the requirements there doesn't appear to be a way to resolve the issue.
Description
I came across this error when installing tensorrt_llm==0.9.0 using poetry on windows. I referred to issue#3050 which described the same failure but was resolved by individually adding modules using poetry. The same steps fail on windows when installing as the dependency manager is unable to find a suitable installation candidate for nvidia-cudnn-cu12. With torch=2.2.0+cu121 already installed, running the following command:
poetry add --source pyindex-nvidia tensorrt-libs==9.3.0.post12.dev1
raises an error: `Package operations: 3 installs, 0 updates, 0 removalsInstalling nvidia-cudnn-cu12 (8.9.2.26): Failed
RuntimeError
Unable to find installation candidates for nvidia-cudnn-cu12 (8.9.2.26)
at C:\Program Files\Python310\lib\site-packages\poetry\installation\chooser.py:74 in choose_for 70│ 71│ links.append(link) 72│ 73│ if not links: → 74│ raise RuntimeError(f"Unable to find installation candidates for {package}") 75│ 76│ # Get the best link 77│ chosen = max(links, key=lambda link: self._sort_key(package, link)) 78│
Cannot install nvidia-cudnn-cu12.`
As far as I looked, there is no nvidia-cuda-cu12(8.9.2.26) for windows platform on nvidia and pytorch indexes.
Environment
TensorRT Version: 9.3.0.post12.dev1
NVIDIA GPU: RTX 4090
NVIDIA Driver Version: 551.52
CUDA Version: V12.2.140
CUDNN Version:
Operating System: Windows OS Build 27552
Python Version (if applicable): 3.10.11
Tensorflow Version (if applicable):
PyTorch Version (if applicable): 2.2.0+cu121
Baremetal or Container (if so, version):
Steps To Reproduce