Как заставить PyTorch использовать ядра CUDA вместо тензорных ядер? ⇐ Python

[Anonymous]

Я хотел бы заставить мою модель использовать ядра CUDA вместо ядер Tensor во время вывода. Есть ли способ специально отключить использование ядра Tensor в PyTorch или заставить операции выполняться только на ядрах CUDA? Кроме того, есть ли какие-либо конкретные настройки в наборе инструментов NVIDIA CUDA или настройках драйвера, которые могут помочь в этом?

Подробнее здесь: https://stackoverflow.com/questions/791 ... nsor-cores