CODEWITHSUNDEEP
tensorflowkerasneural-networkdeep-learningtensor
ramin karimian
ramin karimian

Reputation: 123

How to use tensor cores instead of cuda cores in my code?

I have RTX2070 Nvidia graphic card which has tensor cores on it. I want to run my deep learning code utilizing tensor cores instead of Cuda cores. Is it possible in this graphic card? and isn't there any specific driver that I should install in order to do that? and how should I check to know whether the model is running on tensor cores or Cuda cores? I am using Keras framework on windows 10.

Upvotes: 2

Views: 3603

Answers (1)

carobnodrvo
carobnodrvo

Reputation: 1051

According to NVIDIA:

The container enables Tensor Core math by default

If you want to disable it you can set TF_DISABLE_CUDNN_TENSOR_OP_MATH to 1.

Upvotes: 2

Related Questions