Hey everybody, when I try to run the example from the Flux website for data distributed parallel training , I get the error that the gpu_device is not defined, looking suspiciously familiar to a Lux name. Any suggestions on what I am doing wrong? I am using Julia 1.10.0 and Flux v0.15.2 Here is the…

"gpu_device" not defined in DDP with Flux

Noob December 15, 2024, 2:46pm 2

1 Like

Topic		Replies	Views
Flux.jl: training fails at GPU but works on CPU Machine Learning gpu , flux	1	647	September 19, 2019
Flux.train failing with CUDA GPU Machine Learning	5	933	March 27, 2019
Flux failing on GPU Machine Learning	25	4148	February 21, 2020
Julia Flux error- Unsupported dynamic function invocation (call to pointerref) New to Julia gpu , flux	0	627	January 10, 2021
Lux and Flux GPU function definitions overlap Machine Learning	4	623	October 24, 2022