I have installed CUDA7.5,and because my GPU is GeForce so I use CUDA without CUDNN.After I installed PyTorch followed this web’s command,I found I can’t train the mnist model,even with torch.backends.cudnn.enabled=False.
My GPU’s computation capacity is 2.1, I don’t know if the error reason is this
the error is RuntimeError: cuda runtime error (8) : invalid device function at /data/users/soumith/builder/wheel/pytorch-src/torch/lib/THC/generic/THCTensorMath.cu:15
Can you tell me how to solve this problem?Thank you!