Message Boards Message Boards

GROUPS:

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

Posted 4 months ago
421 Views
|
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
Attachments
Remove
or Discard

Group Abstract Group Abstract