remove number of device (pytorch#127)

* remove number of device

Co-authored-by: Teng Gao <tegao@microsoft.com>

Merge pull request pytorch#84 from gaoteng-git/update_gpu_flag

split is_gpu_used into 3 switches

Fix the setup.py