{"id":23477,"library":"cuequivariance-ops-torch-cu13","title":"cuequivariance-ops-torch (CUDA 13)","description":"GPU-accelerated PyTorch extensions for equivariant neural network primitives, specifically compiled for CUDA 13. Current version 0.10.0, requires Python >=3.10. Part of the cuequivariance ecosystem for efficient group-equivariant operations.","status":"active","version":"0.10.0","language":"python","source_language":"en","source_url":"https://github.com/QUVA-Lab/cuequivariance","tags":["equivariant","torch","CUDA","geometric deep learning","group theory"],"install":[{"cmd":"pip install cuequivariance-ops-torch-cu13","lang":"bash","label":"Install for CUDA 13"}],"dependencies":[{"reason":"PyTorch with CUDA support is required at runtime.","package":"torch","optional":false},{"reason":"Core library for equivariant operations.","package":"cuequivariance","optional":false}],"imports":[{"note":"The package name for import does not include the CUDA suffix; import using 'cuequivariance_ops_torch'.","wrong":"from cuequivariance_ops_torch_cu13 import ...","symbol":"all relevant ops","correct":"import cuequivariance_ops_torch as coe_ops"}],"quickstart":{"code":"import torch\nimport cuequivariance_ops_torch as coe_ops\n\ndevice = torch.device('cuda' if torch.cuda.is_available() else 'cpu')\nx = torch.randn(10, 10, device=device)\ny = coe_ops.some_op(x)  # Replace with actual operation\nprint(y.shape)","lang":"python","description":"Basic usage: import the package (without CUDA suffix) and call an operation on a CUDA tensor."},"warnings":[{"fix":"Use 'import cuequivariance_ops_torch as coe_ops' in code.","message":"Import the package as 'cuequivariance_ops_torch', not with the CUDA suffix ('cuequivariance_ops_torch_cu13'). The suffix is only for pip installation to select the correct CUDA version.","severity":"gotcha","affected_versions":"all"},{"fix":"Match the package suffix (e.g., cu12, cu13) to your installed CUDA toolkit version. Use 'pip install cuequivariance-ops-torch-cu12' for CUDA 12.","message":"CUDA compatibility: The cu13 wheel only works with CUDA 13.x. Installing on a system with a different CUDA version will cause runtime errors (e.g., 'CUDA driver version is insufficient').","severity":"breaking","affected_versions":"0.x"},{"fix":"Ensure your GPU is supported (e.g., Ampere or newer). If not, you may need to build from source or use a CPU fallback.","message":"The package may not be compatible with all CUDA compute capabilities; check the wheel's target architectures (e.g., sm_80, sm_90). Running on an unsupported GPU will raise a 'no kernel image' error.","severity":"gotcha","affected_versions":"all"}],"env_vars":null,"search_vec":"'0.10.0':24 '13':6,21 '3.10':27 'acceler':9 'compil':18 'cuda':5,20,41 'cuequivari':2,31 'cuequivariance-ops-torch':1 'current':22 'deep':43 'ecosystem':32 'effici':34 'equivari':13,37,39 'extens':11 'geometr':42 'gpu':8 'gpu-acceler':7 'group':36,45 'group-equivari':35 'learn':44 'network':15 'neural':14 'op':3 'oper':38 'part':28 'primit':16 'python':26 'pytorch':10 'requir':25 'specif':17 'theori':46 'torch':4,40 'version':23","created_at":"2026-05-01T08:07:59.285535+00:00","updated_at":"2026-05-01T08:07:59.285535+00:00","problems":{"verify_error":"Error while loading libcue_ops.so: libnvrtc.so.13: cannot open shared object file: No such file or directory\nTraceback (most recent call last):\n  File \"<string>\", line 1, in <module>\n  File \"/tmp/tmp5qn_xchb/venv/lib/python3.12/site-packages/cuequivariance_ops_torch/__init__.py\", line 16, in <module"},"ecosystem":"pypi","meta_description":null,"install_score":null,"quickstart_score":null,"quickstart_tag":null,"pypi_latest":"0.10.0","cli_name":"","cli_version":null,"type":"library","homepage":"https://developer.nvidia.com/cuequivariance","github":null,"docs":"https://docs.nvidia.com/cuda/cuequivariance/","changelog":null,"pypi":"https://pypi.org/project/cuequivariance-ops-torch-cu13/","npm":null,"openapi_spec":null,"status_page":null,"smithery":null,"categories":["ai-ml","aws","gcp","azure"],"base_url":null,"auth_type":null,"provenance":{"verified_status":"install_fail","verified_at":"2026-06-29","last_verified":"2026-06-29","next_check":"2026-07-09","install_tag":null}}