Использование GPU для обучения модели

Question

У меня есть небольшое знание использования GPU для обучения модели. Я использую K-средства от scikit-learn для обучения моей модели. Поскольку мои данные очень большие, можно ли обучать эту модель с помощью графического процессора, чтобы сократить время вычислений? или не могли бы вы предложить какие-либо методы использования мощности GPU?

Другой вопрос, если я использую TensorFlow для создания K-средств, как показано в этом блоге.

https://blog.altoros.com/using-k-means-clustering-in-tensorflow.html

Он будет использовать GPU или нет?

Заранее спасибо.

Answer 1

Чтобы проверить, если ваш GPU поддерживает CUDA: https://developer.nvidia.com/cuda-gpus

Scikit-Learn не поддерживается CUDA до сих пор. Вы можете использовать TensorFlow: https://www.tensorflow.org/install/install_linux

Надеюсь, это поможет.

Answer 2

Если у вас есть графический процессор с поддержкой CUDA с Compute Capability 3.0 или выше и установите версию Tensorflow, поддерживающую GPU, то он определенно будет использовать GPU для обучения.

Для дополнения информации о требованиях NVIDIA для запуска TensorFlow с поддержкой GPU проверить по следующей ссылке:

https://www.tensorflow.org/install/install_linux#nvidia_requirements_to_run_tensorflow_with_gpu_support