Hi,
I am getting an error when trying to run your code: "failed to create cublas handle: CUBLAS_STATUS_ALLOC_FAILED". I've used the GPU before, so I am not sure why I am getting this error.
My configuration : Win 10, NVIDIA GeForce 1070, CUDA 9.1, cudnn 7.0.5. I onlt have 1 GPU, so I changed your config file to only train on one gpu (flags.DEFINE_integer('num_gpu', 1, 'number of gpus for distributed training'))
Hi, I am getting an error when trying to run your code: "failed to create cublas handle: CUBLAS_STATUS_ALLOC_FAILED". I've used the GPU before, so I am not sure why I am getting this error.
My configuration : Win 10, NVIDIA GeForce 1070, CUDA 9.1, cudnn 7.0.5. I onlt have 1 GPU, so I changed your config file to only train on one gpu (flags.DEFINE_integer('num_gpu', 1, 'number of gpus for distributed training'))