Test If Cuda Is Working Python Code