Category: Artificial Intelligence

ML-Agents in MacOS with NVIDIA GPU

Mac doesn’t support NVIDIA GPU anymore.

The Titan V can be used, but the driver was retracted right after release, and very unstable.

Using 10XX and Titan Xp is the ‘best’ solution.

MacOS 10.3.5 High Sierra 
Openblas 0.3.7 or 0.3.9 
Latest commandlinetools, and Xcode 10.2
CUDA Toolkit 10.1.243 and customized compiled version of Torch (https://github.com/TomHeaven/pytorch-osx-build)

The most important part is Tom’s customized and compiled Torch .whl files.

Torch 1.7.0 with ml-agents: 0.29.0 works properly.