Hello @111495,
Tt shows that you are using 1.7GB.
I think you executed your script before launchingnvidia-smi, do you confirm that ? If it’s not the case, kill the script that is consuming GPU and run your code.