Running solely on GPU
Required prerequisites
-
Make sure you've read the documentation. Your issue may be addressed there. -
Search the issue tracker and discussions to verify that this hasn't already been reported. +1 or comment there if it has.
What commit version of aidge do you use
-
aidge_core
: 0.0.1 -
aidge_backend_CUDA
: 0.0.1
Problem description
How to run the model on GPU only?
Available backends :
{'cpu', 'cuda'}
Reproducible example code
import aidge_backend_cuda
print("Available backends :")
aidge_core.Tensor.get_available_backends()
Edited by Iustina Ivanova