Install CPU version of torch for CI pipeline

synsense / sinabs

A deep learning library for spiking neural networks which is based on PyTorch, focuses on fast training and supports inference on neuromorphic hardware.

https://sinabs.readthedocs.io

GNU Affero General Public License v3.0

80 stars 8 forks source link

Install CPU version of torch for CI pipeline #115

Closed biphasic closed 1 year ago

biphasic commented 1 year ago

This reduces overall run time of the CI pipeline. This is because normally pip install torch will install CUDA versions by default, but the CI containers don't have GPU access anyway

sheiksadique commented 1 year ago

We are working on a CUDA enabled docker container but I agree with you, we should go with this till then.