Skip to content

Commit c442f0c

Browse files
authored
Add tip for non-CUDA users (#2434)
1 parent 8eb400d commit c442f0c

File tree

1 file changed

+2
-0
lines changed

1 file changed

+2
-0
lines changed

docs/old_tutorials/2024-04-10-blitz.md

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -365,6 +365,8 @@ Seeing our training routine unfold gives us an idea of how the network learnt th
365365

366366
The `gpu` functions you see sprinkled through this bit of the code tell Flux to move these entities to an available GPU, and subsequently train on it. No extra faffing about required! The same bit of code would work on any hardware with some small annotations like you saw here.
367367

368+
If you're not using `CUDA`, be sure to [configure Flux.jl for your GPU backend](https://fluxml.ai/Flux.jl/stable/gpu/#Selecting-GPU-backend).
369+
368370
### Testing the Network
369371

370372
We have trained the network for 100 passes over the training dataset. But we need to check if the network has learnt anything at all.

0 commit comments

Comments
 (0)