search

kanezaki / pytorch-rotationnet

BSD 2-Clause "Simplified" License
80 stars 19 forks source link

RuntimeError: cuda runtime error (11) : invalid argument at /pytorch/aten/src/THC/THCGeneral.cpp:405 #6

Closed longmalongma closed 5 years ago

longmalongma commented 5 years ago

When I try to run your code ,I met this problem,do you know how to solve this problem? File "/home/longma/anaconda2/lib/python2.7/site-packages/torch/nn/modules/conv.py", line 320, in forward self.padding, self.dilation, self.groups) RuntimeError: cuda runtime error (11) : invalid argument at /pytorch/aten/src/THC/THCGeneral.cpp:405

kanezaki commented 5 years ago

I'm not sure but guess that you may have installed pytorch with wrong CUDA version?

longmalongma commented 5 years ago

Think you for your reply,my cuda vision is 9.0,do you know which one version of pytorch can work?

------------------ 原始邮件 ------------------ 发件人: "Asako Kanezaki"notifications@github.com; 发送时间: 2019年6月16日(星期天) 晚上10:52 收件人: "kanezaki/pytorch-rotationnet"pytorch-rotationnet@noreply.github.com; 抄送: "longma"1442342449@qq.com;"Author"author@noreply.github.com; 主题: Re: [kanezaki/pytorch-rotationnet] RuntimeError: cuda runtime error(11) : invalid argument at /pytorch/aten/src/THC/THCGeneral.cpp:405 (#6)

I'm not sure but guess that you may have installed pytorch with wrong CUDA version?

— You are receiving this because you authored the thread. Reply to this email directly, view it on GitHub, or mute the thread.

kanezaki commented 5 years ago

Please try this: conda install pytorch=0.4.1 cuda90 -c pytorch