feinanshan / FANet

MIT License
51 stars 5 forks source link

Torch/Cuda version #2

Closed MaureenZOU closed 4 years ago

MaureenZOU commented 4 years ago

Hi Author, thanks for you work : ) Could you please release the torch and cuda version for this repo. It raise "RuntimeError: cuda runtime error (100) : no CUDA-capable device is detected at /pytorch/aten/src/THC/THCGeneral.cpp:50" error when running the cmd CUDA_VISIBLE_DEVICES=1 python3 speeding.py on torch==1.1.0 and CUDA 10.2. Thanks!