Update docs/src/guide/models/quickstart.md

FluxML · Dec 5, 2024 · 234295e · 234295e
1 parent 3976afd
commit 234295e
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/docs/src/guide/models/quickstart.md b/docs/src/guide/models/quickstart.md
@@ -7,7 +7,8 @@ If you haven't, then you might prefer the [Fitting a Straight Line](overview.md)
 ```julia
 # Install everything, including CUDA, and load packages:
 using Pkg; Pkg.add(["Flux", "CUDA", "cuDNN", "ProgressMeter"])
-using Flux, CUDA, Statistics, ProgressMeter
+using Flux, Statistics, ProgressMeter
+using CUDA  # optional
 device = gpu_device()  # function to move data and model to the GPU
 
 # Generate some data for the XOR problem: vectors of length 2, as columns of a matrix: