Если для обучения модели не хватает собственной видеокарты, покупать RTX ради нескольких экспериментов необязательно. GPU можно арендовать на несколько часов и платить только за время работы.
Один из сервисов для этого — Vast.ai. Здесь можно найти сервер с нужной видеокартой, развернуть на нем окружение с CUDA и PyTorch, подключиться по SSH и…