It is weird 
I disable the cudnn and not i get this error:
RuntimeError: cublas runtime error : the GPU program failed to execute at /opt/conda/conda-bld/pytorch_1512387374934/work/torch/lib/THC/THCBlas.cu:246