notebook.community
Edit and run
First you need to initialize the pyCUDA enviroment, this looks for the NVIDIA GPU's in you sistem and initializes the context to communite whith the GPU
In [ ]: import sys, time, os import numpy as np import pycuda.driver as cuda
import sys, time, os import numpy as np import pycuda.driver as cuda