Closed ycsos closed 2 years ago
In current environment.yml conda not install cuda related package, that lead pytorch choose cpu version instead cuda version. and will fail when python setup.py install or run the infer.py
Thanks! We have updated it in the new pr.
In current environment.yml conda not install cuda related package, that lead pytorch choose cpu version instead cuda version. and will fail when python setup.py install or run the infer.py