4 ms·
From: https://github.com/apple/tensorflow_macos#device-selection-optional https://github.com/apple/tensorflow_macos#device-selection-o... "There is an optional
by subzerofun 6y ago
From: https://github.com/apple/tensorflow_macos#device-selection-optional https://github.com/apple/tensorflow_macos#device-selection-o...
"There is an optional mlcompute.set_mlc_device(device_name=’any') API for ML Compute device selection. The default value for device_name is 'any’, which means ML Compute will select the best available device on your system, including multiple GPUs on multi-GPU configurations."
Hope this is not a stupid question: Does this TF version just work with M1 GPU-Cores / Intels iGPU or also with AMD and supported Nvidia GPUs?
- gok 6y agoIt uses Metal, so the AMD GPUs like the ones in the shipping Mac Pro work.