rGPU - PyTorch tensors on remote GPU

Show HN: Rgpu – a PyTorch device whose tensors live on a remote GPU

rGPU позволяет запускать вычисления PyTorch на удалённой машине с NVIDIA GPU, оставляя код и приложение на вашем ноутбуке. Достаточно указать устройство 'rgpu', и тензоры будут храниться на удалённом сервере, а операции выполняться через TCP. Это решение для тех, у кого нет локальной CUDA, включая пользователей Mac. Также есть CUDA-шим для существующих Linux-программ. Установка через pip, запуск через rgpu-run с SSH-туннелем. Проект с открытым исходным кодом под лицензией Apache 2.0.