how to use the gpu in python