install issue

[sametyle]

Hi, while installing optional requirements for testing, i got the following error. How can i solve this?

sudo pip3 install pytest pytest-pythonpath torch torchvision
Collecting pytest
Using cached pytest-8.3.2-py3-none-any.whl (341 kB)
Collecting pytest-pythonpath
Using cached pytest_pythonpath-0.7.4-py3-none-any.whl (3.7 kB)
Collecting torch
Using cached torch-2.4.0-cp310-cp310-manylinux1_x86_64.whl (797.2 MB)
Collecting torchvision
Using cached torchvision-0.19.0-cp310-cp310-manylinux1_x86_64.whl (7.0 MB)
Collecting pluggy<2,>=1.5
Using cached pluggy-1.5.0-py3-none-any.whl (20 kB)
Collecting exceptiongroup>=1.0.0rc8
Using cached exceptiongroup-1.2.2-py3-none-any.whl (16 kB)
Collecting iniconfig
Using cached iniconfig-2.0.0-py3-none-any.whl (5.9 kB)
Collecting tomli>=1
Using cached tomli-2.0.1-py3-none-any.whl (12 kB)
Collecting packaging
Using cached packaging-24.1-py3-none-any.whl (53 kB)
Collecting pytest
Using cached pytest-6.2.5-py3-none-any.whl (280 kB)
Collecting attrs>=19.2.0
Using cached attrs-24.2.0-py3-none-any.whl (63 kB)
Collecting toml
Using cached toml-0.10.2-py2.py3-none-any.whl (16 kB)
Collecting py>=1.8.2
Using cached py-1.11.0-py2.py3-none-any.whl (98 kB)
Collecting triton==3.0.0
Using cached triton-3.0.0-1-cp310-cp310-manylinux2014_x86_64.manylinux_2_17_x86_64.whl (209.4 MB)
Collecting nvidia-curand-cu12==10.3.2.106
Using cached nvidia_curand_cu12-10.3.2.106-py3-none-manylinux1_x86_64.whl (56.5 MB)
Collecting sympy
Using cached sympy-1.13.2-py3-none-any.whl (6.2 MB)
Collecting nvidia-cusolver-cu12==11.4.5.107
Using cached nvidia_cusolver_cu12-11.4.5.107-py3-none-manylinux1_x86_64.whl (124.2 MB)
Collecting networkx
Using cached networkx-3.3-py3-none-any.whl (1.7 MB)
Collecting nvidia-cuda-nvrtc-cu12==12.1.105
Using cached nvidia_cuda_nvrtc_cu12-12.1.105-py3-none-manylinux1_x86_64.whl (23.7 MB)
Collecting nvidia-cufft-cu12==11.0.2.54
Using cached nvidia_cufft_cu12-11.0.2.54-py3-none-manylinux1_x86_64.whl (121.6 MB)
Collecting nvidia-cusparse-cu12==12.1.0.106
Using cached nvidia_cusparse_cu12-12.1.0.106-py3-none-manylinux1_x86_64.whl (196.0 MB)
Collecting nvidia-nccl-cu12==2.20.5
Using cached nvidia_nccl_cu12-2.20.5-py3-none-manylinux2014_x86_64.whl (176.2 MB)
Collecting nvidia-cuda-runtime-cu12==12.1.105
Using cached nvidia_cuda_runtime_cu12-12.1.105-py3-none-manylinux1_x86_64.whl (823 kB)
Collecting nvidia-nvtx-cu12==12.1.105
Using cached nvidia_nvtx_cu12-12.1.105-py3-none-manylinux1_x86_64.whl (99 kB)
Collecting nvidia-cudnn-cu12==9.1.0.70
Using cached nvidia_cudnn_cu12-9.1.0.70-py3-none-manylinux2014_x86_64.whl (664.8 MB)
Requirement already satisfied: jinja2 in /usr/local/lib/python3.10/dist-packages/jinja2-3.1.4-py3.10.egg (from torch) (3.1.4)
Collecting nvidia-cublas-cu12==12.1.3.1
Using cached nvidia_cublas_cu12-12.1.3.1-py3-none-manylinux1_x86_64.whl (410.6 MB)
Collecting filelock
Using cached filelock-3.15.4-py3-none-any.whl (16 kB)
Collecting nvidia-cuda-cupti-cu12==12.1.105
Using cached nvidia_cuda_cupti_cu12-12.1.105-py3-none-manylinux1_x86_64.whl (14.1 MB)
Collecting fsspec
Using cached fsspec-2024.6.1-py3-none-any.whl (177 kB)
Collecting typing-extensions>=4.8.0
Using cached typing_extensions-4.12.2-py3-none-any.whl (37 kB)
Collecting nvidia-nvjitlink-cu12
Using cached nvidia_nvjitlink_cu12-12.6.20-py3-none-manylinux2014_x86_64.whl (19.7 MB)
Collecting pillow!=8.3.*,>=5.3.0
Using cached pillow-10.4.0-cp310-cp310-manylinux_2_28_x86_64.whl (4.5 MB)
Requirement already satisfied: numpy in /usr/local/lib/python3.10/dist-packages/numpy-2.1.0rc1-py3.10-linux-x86_64.egg (from torchvision) (2.1.0rc1)
Requirement already satisfied: MarkupSafe>=2.0 in /usr/local/lib/python3.10/dist-packages/MarkupSafe-2.1.5-py3.10-linux-x86_64.egg (from jinja2->torch) (2.1.5)
Collecting mpmath<1.4,>=1.1.0
Using cached mpmath-1.3.0-py3-none-any.whl (536 kB)
ERROR: Exception:
Traceback (most recent call last):
File "/usr/lib/python3/dist-packages/pip/_internal/cli/base_command.py", line 165, in exc_logging_wrapper
status = run_func(*args)
File "/usr/lib/python3/dist-packages/pip/_internal/cli/req_command.py", line 205, in wrapper
return func(self, options, args)
File "/usr/lib/python3/dist-packages/pip/_internal/commands/install.py", line 389, in run
to_install = resolver.get_installation_order(requirement_set)
File "/usr/lib/python3/dist-packages/pip/_internal/resolution/resolvelib/resolver.py", line 188, in get_installation_order
weights = get_topological_weights(
File "/usr/lib/python3/dist-packages/pip/_internal/resolution/resolvelib/resolver.py", line 276, in get_topological_weights
assert len(weights) == expected_node_count
AssertionError