CUDA programming in Julia
Documentation | Build Status |
---|---|
The CUDA.jl package is the main programming interface for working with NVIDIA CUDA GPUs using Julia. It features a user-friendly array abstraction, a compiler for writing CUDA kernels in Julia, and wrappers for various CUDA libraries.
The current version of CUDA.jl requires Julia 1.3 or higher, a CUDA-capable GPU with compute capability 3.0 or higher, and an accompanying NVIDIA driver with support for CUDA 10.1 or newer.
The package can be installed with the Julia package manager.
From the Julia REPL, type ]
to enter the Pkg REPL mode and run:
pkg> add CUDA
Or, equivalently, via the Pkg
API:
julia> import Pkg; Pkg.add("CUDA")
For usage instructions and other information, please refer to the documentation at juliagpu.gitlab.io.
The package is tested against, and being developed for, Julia 1.3 and above. Main development and testing happens on Linux, but the package is expected to work on macOS and Windows as well.
Usage questions can be posted on the Julia Discourse forum under the GPU domain and/or in the #gpu channel of the Julia Slack.
Contributions are very welcome, as are feature requests and suggestions. Please open an issue if you encounter any problems.