search

cvlab-stonybrook / SID

Official implementation for ICCV19 "Shadow Removal via Shadow Image Decomposition"
MIT License
99 stars 21 forks source link

CUDA version #8

Closed qpaozixu closed 3 years ago

qpaozixu commented 3 years ago

Hi, thanks for your contribute! I want to know which cuda version is used in this project? I install cuda 10.1 but it show "RuntimeError: CUDA error: invalid device ordinal" when i run python infer.py --model SIDPAMIw --name .\checkout\best_net_G.pth --epoch best

Thanks for any help!