Closed dylanstephen-s closed 4 days ago
Hi @dylanstephen-s, the code in the main branch currently has an issue with CPU fallback on Linux. There is an open pull request #27 that attempts to address the issue. We are still tidying it up and it will be eventually merged to main. I encourage you to clone the code from the open PR and see if it addresses the issue. Feel free to add your feedback there. Duplicate of #26
@dylanstephen-s We have merged #27 to main so you should not need to worry about checking out that branch. Just pull from main and you should be good.
Hello I have been trying to install PROTAX-CPU on an HPC Cluster using a conda env, but I have ran into an issue with the final command: pip install .
I tried with Python version 10, and since it didn't work I noticed on a previous push that it was mentioned to try version 12. Any help would be appreciated.
(protax) [dstephen@gruffalo PROTAX-GPU]$ pip install . Processing /mnt/shared/projects/rbge/dstephen/tools/PROTAX-GPU Preparing metadata (setup.py) ... done Requirement already satisfied: jax in /mnt/shared/scratch/dstephen/apps/conda/envs/protax/lib/python3.12/site-packages (from protax==0.0.0) (0.4.35) Requirement already satisfied: jaxlib in /mnt/shared/scratch/dstephen/apps/conda/envs/protax/lib/python3.12/site-packages (from protax==0.0.0) (0.4.35) Requirement already satisfied: ml-dtypes>=0.4.0 in /mnt/shared/scratch/dstephen/apps/conda/envs/protax/lib/python3.12/site-packages (from jax->protax==0.0.0) (0.5.0) Requirement already satisfied: numpy>=1.24 in /mnt/shared/scratch/dstephen/apps/conda/envs/protax/lib/python3.12/site-packages (from jax->protax==0.0.0) (2.1.3) Requirement already satisfied: opt-einsum in /mnt/shared/scratch/dstephen/apps/conda/envs/protax/lib/python3.12/site-packages (from jax->protax==0.0.0) (3.4.0) Requirement already satisfied: scipy>=1.10 in /mnt/shared/scratch/dstephen/apps/conda/envs/protax/lib/python3.12/site-packages (from jax->protax==0.0.0) (1.14.1) Building wheels for collected packages: protax Building wheel for protax (setup.py) ... error error: subprocess-exited-with-error
× python setup.py bdist_wheel did not run successfully. │ exit code: 1 ╰─> [89 lines of output] running bdist_wheel running build running build_py creating build/lib.linux-x86_64-cpython-312/protax copying src/protax/model.py -> build/lib.linux-x86_64-cpython-312/protax copying src/protax/init.py -> build/lib.linux-x86_64-cpython-312/protax copying src/protax/classify.py -> build/lib.linux-x86_64-cpython-312/protax copying src/protax/baseline.py -> build/lib.linux-x86_64-cpython-312/protax copying src/protax/taxonomy.py -> build/lib.linux-x86_64-cpython-312/protax copying src/protax/protax_utils.py -> build/lib.linux-x86_64-cpython-312/protax creating build/lib.linux-x86_64-cpython-312/knn_jax copying src/knn_jax/knn_jax.py -> build/lib.linux-x86_64-cpython-312/knn_jax copying src/knn_jax/init.py -> build/lib.linux-x86_64-cpython-312/knn_jax running egg_info writing protax.egg-info/PKG-INFO writing dependency_links to protax.egg-info/dependency_links.txt writing requirements to protax.egg-info/requires.txt writing top-level names to protax.egg-info/top_level.txt reading manifest file 'protax.egg-info/SOURCES.txt' adding license file 'LICENSE' writing manifest file 'protax.egg-info/SOURCES.txt' running build_ext -- The CXX compiler identification is GNU 8.5.0 -- Detecting CXX compiler ABI info -- Detecting CXX compiler ABI info - done -- Check for working CXX compiler: /usr/bin/c++ - skipped -- Detecting CXX compile features -- Detecting CXX compile features - done -- Using CMake version 3.30.5 /mnt/shared/projects/rbge/dstephen/tools/PROTAX-GPU -- Found Python: /mnt/shared/scratch/dstephen/apps/conda/envs/protax/bin/python3.12 (found version "3.12.7") found components: Interpreter Development Development.Module Development.Embed -- Performing Test HAS_FLTO -- Performing Test HAS_FLTO - Success -- Found pybind11: /mnt/shared/scratch/dstephen/apps/conda/envs/protax/lib/python3.12/site-packages/pybind11/include (found version "2.13.6") CMake Error at /mnt/shared/scratch/dstephen/apps/conda/envs/protax/share/cmake-3.30/Modules/Internal/CMakeCUDAFindToolkit.cmake:104 (message): Failed to find nvcc.
note: This error originates from a subprocess, and is likely not a problem with pip. ERROR: Failed building wheel for protax Running setup.py clean for protax Failed to build protax ERROR: ERROR: Failed to build installable wheels for some pyproject.toml based projects (protax)