search

garyzhao / SemGCN

The Pytorch implementation for "Semantic Graph Convolutional Networks for 3D Human Pose Regression" (CVPR 2019).
https://arxiv.org/abs/1904.03345
Apache License 2.0
465 stars 78 forks source link

a bug about CUDA #10

Closed ansixwu closed 4 years ago

ansixwu commented 4 years ago

File "/home/dm/anaconda3/envs/py27/lib/python2.7/site-packages/torch/nn/functional.py", line 941, in relu result = torch.relu_(input) RuntimeError: CUDA error: invalid argument

HI is there someone happen for this problem,my environment is ubuntu16.04, pytorch==1.1.0,python=2.7

garyzhao commented 4 years ago

Hi @ansixwu ,

Could you please provide more information about the bug?

Best, Long

ansixwu commented 4 years ago

thanks @garyzhao , I haved solved the bug . because of my cuda version and pytorch version isn't match.