Convolutions does not work on CUDA
Julia Programming Language
Convolutions does not work on CUDA
Hi, I have a problem with Convolutional layers on gpu. It simply does not work. I have installed CUDA and cuDNN. GPU works for “simple” layers like Dense, LayerNorm etc, but when i try evaluate Conv it gives me an error. julia> CUDA.versioninfo() CUDA toolchain: - runtime 13.2.0, artifact installation - unknown driver for 13.2 - compiler 13.3.33, artifact installation CUDA libraries: - cuBLAS: 13.4.0 - cuSPARSE: 12.7.10 - cuSOLVER: 12.2.0 - cuFFT: 12.2.0 - cuRAND: 10.4.2 - CUPTI: 2026.1.1 (...
0 comments
No comments yet.