Обучение делать на полноценном компьютере (куда и будет установлена видеокарта), не на стике.
А использовать уже обученную нейросеть, если вам нужна компактность, другой стик, заточенный под нейросеть - https://software.intel.com/en-us/neural-compute-stick