Probelm with selecting GPU device for model training

[Diego41ITA]

Hello, i'm trying to run the script for model training available on the official docs, but i'm having trouble with selecting the GPU device. I've CUDA installed:

but when i run the script, i get an error:

Can someone help me pls?

[EHxuban11]

Hello @Diego41ITA , thank you for raising this bug, I'm going to address the solution on this issue: #118, please refer there for the solution