Message Boards Message Boards


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

Posted 4 months ago
0 Replies
1 Total Likes

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.

Reply to this discussion
Community posts can be styled and formatted using the Markdown syntax.
Reply Preview
or Discard

Group Abstract Group Abstract