Skip to content

Does this work on amd or other ai card or only with Cuda nvidia card? #247

Answered by nagadomi
lollo1252 asked this question in Q&A
Discussion options

You must be logged in to vote

This project is working with PyTorch.
PyTorch supports ROCm(AMD) on Linux, so this can work with AMD cards on Linux or WSL. #166
macOS/MPS is already supported.
Intel GPU is not yet supported, but it will be possible when PyTorch for xpu is officially released. #232

Regarding performance, I don't know about the performance of cards I don't have. I am not sure that there is anything faster and low cost than RTX3060 Mobile.

Replies: 1 comment 5 replies

Comment options

You must be logged in to vote
5 replies
@nagadomi
Comment options

@lollo1252
Comment options

@lollo1252
Comment options

@nagadomi
Comment options

@beosliege
Comment options

Answer selected by lollo1252
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Category
Q&A
Labels
None yet
3 participants