1
$\begingroup$

(Reported to Wolfram tech support) For example, the command

trained2 = NetTrain[NetModel["LeNet"], "MNIST", TargetDevice -> "GPU"]

hangs, and attempting to abort crashes the kernel (I downloaded and installed all the necessary libraries, including the latest CUDA toolkit (12.9)).

Anyone else having similar issues and/or aware of an available fix? Thanks

Improve this question
$\endgroup$
3
  • $\begingroup$ Haven't downloaded 14.2.1 yet but it runs in 14.2 on Win 11 + NVIDIA RTX 4000 Ada Generation Laptop GPU with CUDA toolkit 12.8 without any issues. If urgent maybe you can install 14.2 and run it. $\endgroup$ Commented May 10 at 3:28
  • 1
    $\begingroup$ Also, have you checked compatibility with Needs["CUDALink`"] and then CUDALink`InstallCUDA`CUDAToolkitCompatibilityInformation[] and CUDAQ[] and CUDAInformation[]? $\endgroup$ Commented May 10 at 3:30
  • $\begingroup$ Currently, the only settings supported by NetTrain are "CPU" and "CUDA" $\endgroup$ Commented May 13 at 15:21

1 Answer 1

Reset to default
4
$\begingroup$

Response from Wolfram Support: "It appears that the new architecture of the RTX 5090 is currently not fully compatible with Wolfram's GPU computation framework. Our developers are aware of the problem and are actively investigating a solution, but unfortunately, [we] cannot provide a specific timeline for a fix at this stage."

Improve this answer
$\endgroup$

Your Answer

By clicking “Post Your Answer”, you agree to our terms of service and acknowledge you have read our privacy policy.

Start asking to get answers

Find the answer to your question by asking.

Ask question

Explore related questions

See similar questions with these tags.