Error 2 example 2

ERROR: LoadError: TaskFailedException
Stacktrace:
 [1] wait
   @ ./task.jl:334 [inlined]
 [2] fetch(t::Task)
   @ Base ./task.jl:349
 [3] top-level scope
   @ ~/knightvision/piece_recognition/KnightVisionServer/test/test_cuda.jl:46

    nested task error: Out of GPU memory trying to allocate 13.184 MiB
    Effective GPU memory usage: 20.20% (2.204 GiB/10.913 GiB)
    Memory pool usage: 30.137 MiB (128.000 MiB reserved)
    Stacktrace:
      [1] macro expansion
        @ ~/.julia/packages/CUDA/0IDh2/src/pool.jl:219 [inlined]
      [2] macro expansion
        @ ./timing.jl:299 [inlined]
      [3] #_alloc#180
        @ ~/.julia/packages/CUDA/0IDh2/src/pool.jl:187 [inlined]
      [4] #alloc#179
        @ ~/.julia/packages/CUDA/0IDh2/src/pool.jl:173 [inlined]
      [5] alloc
        @ ~/.julia/packages/CUDA/0IDh2/src/pool.jl:169 [inlined]
      [6] CUDA.CuArray{Float32, 4, CUDA.Mem.DeviceBuffer}(#unused#::UndefInitializer, dims::NTuple{4, Int64})
        @ CUDA ~/.julia/packages/CUDA/0IDh2/src/array.jl:44
      [7] CuArray
        @ ~/.julia/packages/CUDA/0IDh2/src/array.jl:125 [inlined]
      [8] CuArray
        @ ~/.julia/packages/CUDA/0IDh2/src/array.jl:136 [inlined]
      [9] similar
        @ ./abstractarray.jl:829 [inlined]
     [10] similar
        @ ./abstractarray.jl:828 [inlined]
     [11] similar
        @ ~/.julia/packages/CUDA/0IDh2/src/broadcast.jl:11 [inlined]
     [12] copy
        @ ~/.julia/packages/GPUArrays/gkF6S/src/host/broadcast.jl:47 [inlined]
     [13] materialize(bc::Base.Broadcast.Broadcasted{CUDA.CuArrayStyle{4}, Nothing, typeof(relu), Tuple{Base.Broadcast.Broadcasted{CUDA.CuArrayStyle{4}, Nothing, typeof(+), Tuple{CUDA.CuArray{Float32, 4, CUDA.Mem.DeviceBuffer}, CUDA.CuArray{Float32, 4, CUDA.Mem.DeviceBuffer}}}}})
        @ Base.Broadcast ./broadcast.jl:860
     [14] (::Conv{2, 4, typeof(relu), CUDA.CuArray{Float32, 4, CUDA.Mem.DeviceBuffer}, CUDA.CuArray{Float32, 1, CUDA.Mem.DeviceBuffer}})(x::CUDA.CuArray{Float32, 4, CUDA.Mem.DeviceBuffer})
        @ Flux ~/.julia/packages/Flux/BPPNj/src/layers/conv.jl:166
     [15] applychain(fs::Tuple{Conv{2, 4, typeof(relu), CUDA.CuArray{Float32, 4, CUDA.Mem.DeviceBuffer}, CUDA.CuArray{Float32, 1, CUDA.Mem.DeviceBuffer}}, MaxPool{2, 4}, Conv{2, 4, typeof(relu), CUDA.CuArray{Float32, 4, CUDA.Mem.DeviceBuffer}, CUDA.CuArray{Float32, 1, CUDA.Mem.DeviceBuffer}}, MaxPool{2, 4}, typeof(flatten), Dense{typeof(relu), CUDA.CuArray{Float32, 2, CUDA.Mem.DeviceBuffer}, CUDA.CuArray{Float32, 1, CUDA.Mem.DeviceBuffer}}, Dense{typeof(relu), CUDA.CuArray{Float32, 2, CUDA.Mem.DeviceBuffer}, CUDA.CuArray{Float32, 1, CUDA.Mem.DeviceBuffer}}, Dense{typeof(identity), CUDA.CuArray{Float32, 2, CUDA.Mem.DeviceBuffer}, CUDA.CuArray{Float32, 1, CUDA.Mem.DeviceBuffer}}}, x::CUDA.CuArray{Float32, 4, CUDA.Mem.DeviceBuffer})
        @ Flux ~/.julia/packages/Flux/BPPNj/src/layers/basic.jl:47
     [16] Chain
        @ ~/.julia/packages/Flux/BPPNj/src/layers/basic.jl:49 [inlined]
     [17] inference(imgs::CUDA.CuArray{Float32, 4, CUDA.Mem.DeviceBuffer})
        @ Main ~/knightvision/piece_recognition/KnightVisionServer/test/test_cuda.jl:19
     [18] (::var"#13#14")()
        @ Main ./threadingconstructs.jl:178
in expression starting at /home/mattias/knightvision/piece_recognition/KnightVisionServer/test/test_cuda.jl:43