What is your cuda version?

GOATmessi8 / ASFF

yolov3 with mobilenet v2 and ASFF

GNU General Public License v3.0

1.05k stars 216 forks source link

What is your cuda version? #81

Closed Jucjiaswiss closed 4 years ago

Jucjiaswiss commented 4 years ago

I got this error when run demo.py, import DCN ImportError: libcudart.so.10.1: cannot open shared object file: No such file or directory I installed cuda-10.0, and install pytorch with command conda install pytorch=1.3.1 torchvision cudatoolkit=10.0 -c pytorch, anyone know the reason?

Jucjiaswiss commented 4 years ago

Problem solved. I removed the previous env and start a new one by reinstall pytorch. conda install pytorch=1.3.1 torchvision cudatoolkit=10.0 -c pytorch same command here but it works this time.