I would like to train a sparse neural network using Flux. I can train it in CPU, but I would like to do it using GPU, profiting from CuSparseMatrixCSC. However, when I try to run the code below I get ERROR: LoadError: This object is not a GPU array. Is it possible to train sparse networks in GPU? …

Is it possible to train sparse neural networks in GPU?

ChrisRackauckas June 17, 2022, 2:57am 4

Yes, make an MWE about reshaping sparse CUDA matrices on CUDA.jl and it can get fixed up.

1 Like