From 7eb82a90baeb0df7869b8c3656422d5d7fcc98cf Mon Sep 17 00:00:00 2001 From: Axel Huebl Date: Mon, 17 Oct 2022 15:31:27 -0700 Subject: [PATCH] README: List CUDA Compatible Test Packages --- README.md | 5 +++++ 1 file changed, 5 insertions(+) diff --git a/README.md b/README.md index 9918cca6..c6f7143f 100644 --- a/README.md +++ b/README.md @@ -64,6 +64,11 @@ Optional dependencies include: - further [optional dependencies of AMReX](https://github.com/AMReX-Codes/amrex/) - [pytest](https://docs.pytest.org/en/stable/) 6.2+: for running unit tests +Optional CUDA-capable dependencies for tests include: +- [cupy](https://github.com/cupy/cupy#installation) 11.2+ +- [numba](https://numba.readthedocs.io/en/stable/user/installing.html) 0.56+ +- [torch](https://pytorch.org/get-started/locally/) 1.12+ + ### Install Dependencies macOS/Linux: