Turicreate: Style Transfer... Mac GPU option?

[enzyme69]

I am testing turicreate==5.0b1 following your video at WWDC 2018, and trying to make "Style Transfer" model. When executed and processing, it seems like it is utilizing CPU by default. Is there a way to force use GPU? Thanks!

Using CPU, 1000 iterations is around 10+ hours on my MacBookPro 2017 with Touch Bar.

Similar to: https://github.com/apple/turicreate/issues/630

[enzyme69]

Yes MacOS 10.14 beta is installed. Will try now cheers~

[enzyme69]

Still not automatically utilizing GPU, maybe my MacBookPro does not have proper GPU?

SYSTEM:

Hardware Overview:

Model Name: MacBook Pro
Model Identifier: MacBookPro14,3

MacOS:
10.14 Beta (18A293u)
Radeon Pro 555:

Chipset Model: Radeon Pro 555
Type: GPU
Bus: PCIe
PCIe Lane Width: x8
VRAM (Dynamic, Max): 2048 MB
Vendor: AMD (0x1002)
Device ID: 0x67ef
Revision ID: 0x00c7
ROM Revision: 113-C980AJ-927
VBIOS Version: 113-C9801AP-A02
EFI Driver Version: 01.00.927
Automatic Graphics Switching: Supported
gMux Version: 4.0.29 [3.2.8]
Metal: spdisplays_metalfeaturesetfamily21
Displays:
Colour LCD:
Display Type: Built-In Retina LCD
Resolution: 2880 x 1800 Retina
Framebuffer Depth: 30-Bit Colour (ARGB2101010)
Main Display: Yes
Mirror: Off
Online: Yes
Rotation: Supported
Automatically Adjust Brightness: No

[enzyme69]

I am sure I got all the latest beta. Does it need Python 3.6? I am using 3.5 with Jupyter Notebook, maybe Conda Install.

[srikris]

Sorry for the late response. Style transfer is currently CPU only on Mac. We will be working to add GPU support to it over time. Sorry for the trouble.

[tankfly2014]

nv 1060 6GB ram. or 1080.
CPU is dead road.

[josefgrunig]

Any roadmap/deadline for this feature?
The style transfer task is useless without GPU acceleration

[nickjong]

Yeah, the current implementation is really designed around MXNet, which is only GPU accelerated on Linux with Nvidia. We're still working through a Metal Performance Shaders implementation for macOS, but this at least requires some macOS-level blocking issues to be resolved....

[josefgrunig]

Thank you for the answer, meanwhile I got it working on Google Colab with GPU acceleration.

[AugustRush]

any update?

[syoutsey]

Hi @enzyme69, this feature was released as part of Turi Create 5.8. We'll close this issue and if you see issues please create a new issue. Thanks!