fix: try fixing cuda install in tests (#1126)

Author: avik-pal

Project.toml

@@ -83,7 +83,7 @@ Flux = "0.15"
 ForwardDiff = "0.10.36"
 FunctionWrappers = "1.1.3"
 Functors = "0.5"
-GPUArraysCore = "0.1.6, 0.2"
+GPUArraysCore = "0.1.6"
 LinearAlgebra = "1.10"
 LossFunctions = "0.11.1, 1"
 LuxCore = "1.2"