web-dev-qa-db-fra.com

Google Tensorflow prend-il en charge OpenCL

Google Tensorflow prend-il en charge OpenCL ... ou n'est-ce toujours que Cuda?

OpenCL ne semble pas encore être pris en charge (avril 2017) par ce problème ouvert - https://github.com/tensorflow/tensorflow/issues/22 mais je continue à lire que le support existe - je pourrais manquer quelque chose.

10
NumberCrunchy55

tf-coriander est une implémentation de Tensorflow pour les GPU OpenCL 1.2. Il est basé sur coriandre , qui est un compilateur général pour exécuter le code NVIDIA® CUDA ™ sur les appareils OpenCL 1.2. Divulgation: je suis l'auteur des deux projets.

17
Hugh Perkins

Il existe un support OpenCL via SYCL sur Tensorflow, certaines fonctionnalités sont disponibles, d'autres sont en cours: https://github.com/tensorflow/tensorflow/issues/22#issuecomment-266050835

5
Ruyk