3 ms·
> Riff doesn't track which GPU you use at this time, so it can't track that which build of `pytorch` to use, sorry. It might be something we make it aware of if
by gpm 4y ago
> Riff doesn't track which GPU you use at this time, so it can't track that which build of `pytorch` to use, sorry. It might be something we make it aware of if we have enough evidence it would be helpful.
Would it be possible to do something like make it depend on a cargo feature flag, or something similar?
It's probably better that this isn't automatic, since the correct choice of device isn't always obvious just based on what is available.