Python Code To Test Gpu