Failed to create cublas handle - CUBLAS_STATUS_ALLOC_FAILED

[anburiyan]

I tried to run the code using ImageNet weights. Getting the error "Failed to create cublas handle - CUBLAS_STATUS_ALLOC_FAILED" while starting epoch.

Below are the my NVIDEA configuration:
nvcc: NVIDIA (R) Cuda compiler driver
Copyright (c) 2005-2017 NVIDIA Corporation
Built on Fri_Sep__1_21:08:32_Central_Daylight_Time_2017
Cuda compilation tools, release 9.0, V9.0.176
GPU Mem : 2GB

Can you suggest why this issue is appearing? Need to modify the config file to adequate with my GPU memory

[markjay4k]

I'm not exactly sure what the error is.