tsai icon indicating copy to clipboard operation
tsai copied to clipboard

using GPU on m1 Pro Mac

Open AsafShul opened this issue 3 years ago • 0 comments

hey, first of all thanks for this great repo!

Im using a MacBook with M1 pro chip,

I have installed PyTorch with the new GPU support, and it seems to be working - when running this commands I get True, True " # this ensures that the current MacOS version is at least 12.3+ print(torch.backends.mps.is_available())

# this ensures that the current current PyTorch installation was built with MPS activated.
print(torch.backends.mps.is_built())

"

But running the tsai library only utilize the CPU on my machine,

How can I fix this? thanks

AsafShul avatar Jul 28 '22 07:07 AsafShul