CuPy acts as a drop-in replacement to run existing NumPy/SciPy code on NVIDIA CUDA or AMD ROCm platforms. CUDA 11.x (11.2+) x86_64 / aarch64 pip install cupy-cuda11x CUDA 12.x x86_64 / aarch64 pip ...
This repository contains Starlark implementation of CUDA rules in Bazel. These rules provide some macros and rules that make it easier to build CUDA with Bazel. Enable or disable all rules_cuda ...