learex-zz / macOS-eGPU

Make your Mac compatible with NVIDIA and AMD eGPUs. (macOS High Sierra)
Other
691 stars 53 forks source link

Has anyone installed TensorFlow for GPU? #64

Open cfox570 opened 5 years ago

cfox570 commented 5 years ago

TensorFlow requires Nvidia GPUs. Has anyone used this script to enable TensorFlow GPU running on a Mac? What type of performance gain?

jayhxmo commented 5 years ago

I tried, but it didn't work out and I ended up building a separate PC for running Tensorflow. I'm only a beginner in the field so I could have gotten things wrong but my understanding was that Tensorflow does not support Nvidia GPUs anymore on the macOS packages.

hwiorn commented 5 years ago

I am using aorus 1080 egpu box with MBP2013(macOS 10.13.6) well.

Official tensorflow does not support GPU on macOS anymore. But custom build can do.

I recommend this tensorflow pre-build release.

If you want pytorch pre-build, Check his pytorch repo