Added warning if the memory is too large and updated CUDA version.