AlexNet RuntimeError: CUDA error: CUBLAS_STATUS_ALLOC_FAILED when calling `cublasCreate(handle)`

No it is completely fine I’m just here to help people I’m not doing much else right now. So after this line

inputs = inputs/1600
do this 
print(inputs.shape)

That will output the shape. Also why are you dividing it by 1600?