Add note about loading CUDA.jl to the docs

Evovest · Oct 5, 2023 · 3a3fc7f · 3a3fc7f
1 parent 8a42cef
commit 3a3fc7f
Showing 1 changed file with 6 additions and 0 deletions.
diff --git a/docs/src/index.md b/docs/src/index.md
@@ -68,6 +68,12 @@ m = fit_evotree(config, dtrain; target_name="y", fnames=["x1", "x3"]);
 
 ### GPU Acceleration
 
+EvoTrees supports training and inference on Nvidia GPU's with [CUDA.jl](https://github.com/JuliaGPU/CUDA.jl).
+Note that on Julia ≥ 1.9 CUDA support is only enabled when CUDA.jl is installed and loaded, by another package or explicitly with e.g.
+```julia
+using CUDA
+```
+
 If running on a CUDA enabled machine, training and inference on GPU can be triggered through the `device` kwarg: 
 
 ```julia