NNgen / nngen

NNgen: A Fully-Customizable Hardware Synthesis Compiler for Deep Neural Network
Apache License 2.0
340 stars 46 forks source link

install issue #67

Open sametyle opened 3 months ago

sametyle commented 3 months ago

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