Hi @jasg, did you Build from Source 228? You probably need to build from the source since PyTorch is installed without CUDA support by default on MacOS. This is due to CUDA being a Nvidia proprietary technology.