MXNet supports mx.cpu() for CPU and mx.gpu(0) , mx.gpu(1) , etc for GPU. You can specify context when creating NDArray: # creates on CPU (the default).
確定! 回上一頁