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

1 次查看(过去 30 天)
How to use Titan-V tensor cores from Matlab to Deep Learning and General Purpose calculations

采纳的回答

Krishna Bindumadhavan
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:
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.

更多回答(0 个)

类别

Help CenterFile Exchange 中查找有关 Deep Learning Toolbox 的更多信息

Community Treasure Hunt

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

Start Hunting!

Translated by