Hello, good evening. Im having an issue trying to run the Flux get started example. Always got ArgumentError: Illegal conversion of a CUDA.DeviceMemory to a Ptr{Float32} It brokes in line 17: out1 = model(noisy |> device) # 2x1000 Matrix{Float32}, or CuArray{Float32} Source code: # Inst…

Help with CUDA and Flux. DeviceMemory issue

mohamed.d180 February 2, 2025, 10:52pm 2

I can confirm that every thing works OK, here is a screen shot from Google colab, Please check your CUDA installation using

using CUDA
CUDA.versioninfo()

1 Like

Topic		Replies	Views
Flux.train failing with CUDA GPU Machine Learning	5	933	March 27, 2019
`CUDA error: out of memory` with Flux Machine Learning flux	4	1668	August 24, 2020
Flux's model-zoo CIFAR10 example saturates 8GB gpu General Usage gpu , flux	5	672	June 29, 2020
Flux.jl: training fails at GPU but works on CPU Machine Learning gpu , flux	1	647	September 19, 2019
Flux failing on GPU Machine Learning	25	4148	February 21, 2020