CUDA working but NetTrain[~,"TargetDevice"->"GPU"] doesn't

Am I supposed to do a different command within NetTrain[] to use CUDA or is “TargetDevice”->“GPU” still supposed to work? (CUDA examples evaluate successfully.)

The net gets trained but CPU spikes to 56% from a baseline of 1% while the NetTrain is evaluated and GPU does not seem to get used.

Gfx: Nvidia 2080Ti
CPU: AMD 3950x
Wolfram v. 12.2

Never mind: I think it is working now.

1 Like