Skip to content

Commit

Permalink
README: List CUDA Compatible Test Packages
Browse files Browse the repository at this point in the history
  • Loading branch information
ax3l committed Oct 17, 2022
1 parent 43e35e7 commit 7eb82a9
Showing 1 changed file with 5 additions and 0 deletions.
5 changes: 5 additions & 0 deletions README.md
Original file line number Diff line number Diff line change
Expand Up @@ -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:
Expand Down

0 comments on commit 7eb82a9

Please sign in to comment.