When Colaboratory opens, first change the setting to use GPU. Click 'Change runtime type' in the 'Runtime' menu. Set 'Hardware Acceleration' to 'GPU' and click 'Save'. Enter the Python code here.