search

HKUST-Aerial-Robotics / Stereo-RCNN

Code for 'Stereo R-CNN based 3D Object Detection for Autonomous Driving' (CVPR 2019)
MIT License
690 stars 177 forks source link

THCudaCheck FAIL #25

Open chenzhanjie opened 5 years ago

chenzhanjie commented 5 years ago

I run the project in pytorch 1.0 version, and then the result have been print. But there is a problem in the terminal: THCudaCheck FAIL file=/opt/conda/conda-bld/pytorch_1544174967633/work/aten/src/THC/THCGeneral.cpp line=405 error=11 : invalid argument I did not find the answer, so I hope that I can get some help at here. Thank you.

PeiliangLi commented 5 years ago

Have you installed cuda successfully?

chenzhanjie commented 5 years ago

Have you installed cuda successfully?

Yes, i created an environment in Anaconda3, and then i installed the pytorch1.0 with cuda9.0 and python3.6.

PeiliangLi commented 5 years ago

What is your output of nvidia-smi, nvcc --version, and python import torch torch.__version__?

chenzhanjie commented 5 years ago

What is your output of nvidia-smi, nvcc --version, and python import torch torch.__version__?

these output you mentioned are right, but now i think the problem may be caused by network connection, I am in China. So just like you said,it is possible that i haven't installed it completely. But I do not find which part i missed yet. So I am trying to reinstall all the components once.

wanglaotou commented 4 years ago

i met the same error when i run python demo.py. and my nvcc -V is v10.0(titan rtx), and my torch version is 1.0.0, torchvision is 0.2.2. should the titan rtx install with cuda10.0? like conda install pytorch=1.0.0 cuda100 -c pytorch

wanglaotou commented 4 years ago

after run conda install pytorch=1.0.0 cuda100 -c pytorch and conda install torchvision -c pytorch, it worked for me, wish it may help you : )