Problem Checking CUDA

[JNaranjo-Alcazar]

Hi everyone!

Following the instructions in the repo, the output of the code is:

UserWarning: volatile was removed and now has no effect. Use `with torch.no_grad():` instead.
  inputs = Variable(inputs, volatile=True)
THCudaCheck FAIL file=/opt/conda/conda-bld/pytorch_1544174967633/work/aten/src/THC/THCGeneral.cpp line=405 error=11 : invalid argument

Anyone has faced the same problem?

I tried the solution in this issue pytorch/pytorch#15797

using pip install -U https://download.pytorch.org/whl/cu100/torch-1.0.0-cp36-cp36m-linux_x86_64.whl but it did not work.
Thanks in advance!