How to use Titan-V tensor cores from Matlab to Deep Learning and General Purpose calculations

Question

Alexander Sonin 2018-2-12

0
链接

此问题的直接链接

https://ww2.mathworks.cn/matlabcentral/answers/382154-how-to-use-titan-v-tensor-cores-from-matlab-to-deep-learning-and-general-purpose-calculations

回答： Krishna Bindumadhavan 2019-9-14

采纳的回答： Krishna Bindumadhavan

How to use Titan-V tensor cores from Matlab to Deep Learning and General Purpose calculations

0 个评论
显示 -2更早的评论隐藏 -2更早的评论

请先登录，再进行评论。

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

Answer 1

Krishna Bindumadhavan 2019-9-14

0
链接

此回答的直接链接

https://ww2.mathworks.cn/matlabcentral/answers/382154-how-to-use-titan-v-tensor-cores-from-matlab-to-deep-learning-and-general-purpose-calculations#answer_391855

You may take advantage of tensor cores after generating code for a trained neural network with GPU Coder. You can take a look at the Deep Learning Prediction with NVIDIA TensorRT example but set the DataType property of the DeepLearningConfig to 'fp16'. This will use the Tensor cores on the Titan-V.

In MATLAB, there is support for the half precision data type via the half precision object in the fixed point designer toolbox:

https://www.mathworks.com/help/fixedpoint/ref/half.html.

In addition, general code generation support for fp16 data type via MATLAB Coder and GPU coder is under active development and can be expected in a future release.