The CUDA.jl package is the main entrypoint for for programming NVIDIA GPUs using CUDA. The package makes it possible to do so at various abstraction levels, from easy-to-use arrays down to hand-written kernels using low-level CUDA APIs.
The Julia CUDA stack requires a functional CUDA-setup, which includes both a driver and matching toolkit. Once you've set that up, continue by installing the CUDA.jl package:
using Pkg Pkg.add("CUDA")
To make sure everything works as expected, try to load the package and if you have the time execute its test suite:
using CUDA using Pkg Pkg.test("CUDA")
For more details on the installation process, consult the Installation section. To understand the toolchain in more detail, have a look at the tutorials in this manual. It is highly recommended that new users start with the Introduction tutorial. For an overview of the available functionality, read the Usage section. The following resources may also be of interest:
The Julia CUDA stack has been a collaborative effort by many individuals. Significant contributions have been made by the following individuals:
- Tim Besard (@maleadt) (lead developer)
- Valentin Churavy (@vchuravy)
- Mike Innes (@MikeInnes)
- Katharine Hyatt (@kshyatt)
- Simon Danisch (@SimonDanisch)
Much of the software in this ecosystem was developed as part of academic research. If you would like to help support it, please star the repository as such metrics may help us secure funding in the future. If you use our software as part of your research, teaching, or other activities, we would be grateful if you could cite our work. The CITATION.bib file in the root of this repository lists the relevant papers.