Open superQCman opened 4 months ago
Traceback (most recent call last):
File "main.py", line 155, in <module>
torch.cuda.set_device(0)
File "main.py", line 142, in main
for epoch in range(1, args.epochs + 1):
File "main.py", line 44, in train
optimizer.zero_grad()
File "/home/sim/anaconda3/envs/gpgpu-sim/lib/python3.6/site-packages/torch/nn/modules/module.py", line 477, in __call__
result = self.forward(*input, **kwargs)
File "main.py", line 24, in forward
x=x.float()
File "/home/sim/anaconda3/envs/gpgpu-sim/lib/python3.6/site-packages/torch/nn/modules/module.py", line 477, in __call__
result = self.forward(*input, **kwargs)
File "/home/sim/anaconda3/envs/gpgpu-sim/lib/python3.6/site-packages/torch/nn/modules/conv.py", line 301, in forward
self.padding, self.dilation, self.groups)
RuntimeError: cublas runtime error : the GPU program failed to execute at /home/sim/pytorch-gpgpusim/aten/src/THC/THCBlas.cu:249
makefile:7: recipe for target 'run-main' failed
make: *** [run-main] Error 1
After I disabled cudnn, the Error changed to this
i found this, https://github.com/gpgpu-sim/pytorch-gpgpu-sim/issues/5
How to solve this problem? I use cuda 10.1, cudnn 7.6.5, python 3.6