Tensor Cores and deep learning

6 次查看（过去 30 天）
Gerardo Suarez
Gerardo Suarez 2019-6-24
评论： Alexander Sonin 2021-12-3
Dear all
I have a GeForce RTX 2060. How can I take advantage of its 240 tensor cores for deep learning? I have read that I should set the precision to FP16, but using the "trainnetwork" function there does not seem to be any option.
Thank you

请先登录，再回答此问题。

采纳的回答

Krishna Bindumadhavan
编辑：Krishna Bindumadhavan 2019-9-14
Currently, there is no support for training a neural network with half precision data type. This is expected in a future release.
You can however take advantage of tensor cores, while generating code for deploying a trained neural network with GPU Coder. You can look at Deep Learning Prediction with NVIDIA TensorRT example , but set the DataType property of the DeepLearningConfig to 'fp16'. This will use the Tensor cores on a Volta or Turing card such as the RTX series for inference.
In addition, MATLAB supports the half data type via the half precision object in the fixed point designer toolbox: https://www.mathworks.com/help/fixedpoint/ref/half.html?s_tid=doc_ta. General code generation support for the half type, via MATLAB Coder and GPU coder is under active development and can be expected in an upcoming release.
  1 个评论
Alexander Sonin
Alexander Sonin 2021-12-3
Krishna kept silent about the exact answer to this question: MATLAB cannot use tensor cores to train neural networks !!!

请先登录，再进行评论。

更多回答（0 个）

类别

在 Help Center 和 File Exchange 中查找有关 Deep Learning Toolbox 的更多信息

标签

产品


版本

R2018b

Community Treasure Hunt

Find the treasures in MATLAB Central and discover how the community can help you!

Start Hunting!

Translated by