Hello, I have been trying to implement Opacus to my models optimiser. I have run it on colab. At the time of training I run into this error.
RuntimeError: CUDA error: CUBLAS_STATUS_ALLOC_FAILED when calling cublasCreate(handle)
Could someone please guide me on this.
Thanks!
Screen Shot 2020-10-09 at 11.03.33 AM859×608 64.1 KB