Use GPU system in lava-dl CI
bamsumit opened this issue · comments
User story
As a user, I want to run lava-dl CI on a GPU system to test GPU execution of the code. Lava-dl training codes are intended to run on GPU. There are specific code segments that only run on GPU which are not captured by the current CI
Conditions of satisfaction
- Poetry handles the installation of correct torch version compatible with the cuda driver and runtime in the system.
- CI runs on a linux system NVIDIA GPU, in addition to a CPU only system.