WOLFRAM

What are the minimum requirements for GPU training of neural networks in Wolfram Language?


Please note that CUDALink and GPU training of neural networks are separate features with separate requirements. If you are looking for the CUDALink requirements, please consult the appropriate article.

If your machine has an NVIDIA GPU, the necessary paclet will be downloaded automatically the first time you use TargetDevice -> “GPU” in a NetTrain or other net evaluation.

Your NVIDIA GPU must be compatible with a supported CUDA Toolkit version, and the NVIDIA driver needs to be up to date.

Wolfram Language Version NVIDIA Driver Version Compute Capability
Linux Windows macOS
15.0 570.125 572.61 N/A 5.0
14.3 570.125 572.61 N/A 5.0
14.2 510.47 511.65 N/A 3.7
14.1 510.47 511.65 N/A 3.7
14.0 510.47 511.65 N/A 3.7
13.3 510.47 511.65 N/A 3.7
13.2 495.29 496.04 N/A 3.7
13.1 495.29 496.04 N/A 3.7
13.0 470.42 471.41 N/A 3.7
12.3 460.27 460.82 N/A 3.7
12.2 460.27 460.82 N/A 3.7
12.1 418.39 418.96 418.105 3.7
12.0 410.48 411.31 410.130 3.0
11.3 390.46 391.29 387.128 3.0

GPU training of neural networks is no longer supported on macOS starting with Wolfram Language 12.2, as Apple dropped macOS support of CUDA.

For more compatibility information, please see NVIDIA’s CUDA compatibility documentation.

In case of any inconsistency with their information, please contact NVIDIA.

Is this article helpful?
Yes
No

Any comments?

Thank you for your feedback.

Submit

Contact Support

Whether you have a question about billing, activation or something more technical, we are ready to help you.

1-800-WOLFRAM (+1-217-398-0700 for international callers)

Customer Support

Monday–Friday
8am–5pm US Central Time

  • Product registration or activation
  • Pre-sales information and ordering
  • Help with installation and first launch

Advanced Technical Support (for eligible customers)

Monday–Thursday
8am–5pm US Central Time

Friday
8:30–10am & 11am–5pm US Central Time

  • Priority technical support
  • Product assistance from Wolfram experts
  • Help with Wolfram Language programming
  • Advanced installation support