try_gpu() function for macos #2655
Description
Hi, I am trying to use d2l on MacOS. I found that try_gpu() function can only detect 'cuda', but not 'mps'. Can anyone add this?
Thanks
{{ message }}
Hi, I am trying to use d2l on MacOS. I found that try_gpu() function can only detect 'cuda', but not 'mps'. Can anyone add this?
Thanks