You signed in with another tab or window. Reload to refresh your session.You signed out in another tab or window. Reload to refresh your session.You switched accounts on another tab or window. Reload to refresh your session.Dismiss alert
Traceback (most recent call last):
File "/root/Lee_pro/gcn_v2/cuda.py", line 10, in
c = a.matmul(b)
RuntimeError: cublas runtime error : the GPU program failed to execute at /pytorch/aten/src/THC/THCBlas.cu:450
env : pytorch1.1-cuda9-cudnn7-ubuntu16.04
I run this on my school server (docker env)
The text was updated successfully, but these errors were encountered:
import torch
torch.cuda.set_device(0)
a = torch.randn((2,3,6)).cuda()
b = torch.randn((2,6,7)).cuda()
c = a.matmul(b)
print("success")
Error :
Traceback (most recent call last):
File "/root/Lee_pro/gcn_v2/cuda.py", line 10, in
c = a.matmul(b)
RuntimeError: cublas runtime error : the GPU program failed to execute at /pytorch/aten/src/THC/THCBlas.cu:450
env : pytorch1.1-cuda9-cudnn7-ubuntu16.04
I run this on my school server (docker env)
The text was updated successfully, but these errors were encountered: