add cuda environment for conda environment.yml

[ycsos]

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

[oahzxl]

Thanks! We have updated it in the new pr.