Pytorch supports GPU accelerated training and interference on Macs by using Metal now. This increases the training speed significantly, especially on Apple Silicon Macs. It would be great if this package would automatically use the "mps" device if available, such as it does with CUDA.
Pytorch supports GPU accelerated training and interference on Macs by using Metal now. This increases the training speed significantly, especially on Apple Silicon Macs. It would be great if this package would automatically use the "mps" device if available, such as it does with CUDA.