我有 RTX2070 Nvidia 显卡，上面有张量核心。我想利用张量核心而不是 Cuda 核心来运行我的深度学习代码。这个显卡可以吗？为了做到这一点，我是否应该安装任何特定的驱动程序？ 我应该如何检查模型是否在张量核心或 Cuda 核心上运行？ 我在 Windows 10 上使用 Keras 框架。
最佳答案
根据NVIDIA :
The container enables Tensor Core math by default
如果您想禁用它，可以将 TF_DISABLE_CUDNN_TENSOR_OP_MATH 设置为 1。
关于tensorflow - 如何在我的代码中使用张量核心而不是 cuda 核心？，我们在Stack Overflow上找到一个类似的问题： https://stackoverflow.com/questions/57816915/