Hi there. This is much above my knowledge level, so pardon me if I’m at all off the mark here. As I understand it, Argos supports CUDA, which is limited to certain Nvidia cards. There is a project called HIP that may open up support to AMD cards as well. Is this something that Argos could benefit…

Support for GPU acceleration on AMD with HIP?

argosopentech August 14, 2022, 2:05pm 2

I’m not sure about HIP specifically but you should be able to use any hardware Pytorch supports.