Message Boards Message Boards

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

Posted 3 years ago

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.

POSTED BY: Renay Oshop
Reply to this discussion
Community posts can be styled and formatted using the Markdown syntax.
Reply Preview
Attachments
Remove
or Discard

Group Abstract Group Abstract