messing with CUDA in python