Skip to content

Commit 601b70b

Browse files
Merge pull request #971 from ZXEcoder/patch-1
Update GPUs.md
2 parents 3e2c350 + 7c73390 commit 601b70b

File tree

1 file changed

+2
-0
lines changed

1 file changed

+2
-0
lines changed

docs/src/examples/GPUs.md

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -75,6 +75,8 @@ same code works on CPUs and GPUs, dependent on `using LuxCUDA`.
7575
using Lux, Optimization, OptimizationOptimisers, Zygote, OrdinaryDiffEq, Plots, LuxCUDA,
7676
SciMLSensitivity, Random, ComponentArrays
7777
import DiffEqFlux: NeuralODE
78+
const cdev = cpu_device()
79+
const gdev = gpu_device()
7880
7981
CUDA.allowscalar(false) # Makes sure no slow operations are occurring
8082

0 commit comments

Comments
 (0)