Re:Apple M1 support for TensorFlow 2.5 pluggable device API
by truth_ on Wednesday June 09, 2021 @05:56AM
(#27446142)
PyTorch currently provides RoCM support. Has anyone tried that?
[ Parent
| on HN ]
Re:Apple M1 support for TensorFlow 2.5 pluggable device API
by yza on Wednesday June 09, 2021 @06:17AM
(#27446251)
I tried it on a Radeon VII. It is little hard to get running as it does not work with latest kernel, but otherwise it kind of works with some quirks. One of the quirks is the first epoch of your training is very slow to start compared to Nvidia cards. Also, the speed of training is slow.