Limin

Results 4 comments of Limin

1. Change default parameters of --device to “cuda device” ![image](https://user-images.githubusercontent.com/29683145/132706958-74a68ca4-ee12-415f-bca2-1902ed3211b0.png) 2. Run the demo.py ![image](https://user-images.githubusercontent.com/29683145/132707181-8f338703-1ed5-471c-9032-09c0b370af03.png)

If you print the 'torch.cuda.is_available()' before select_device function ![image](https://user-images.githubusercontent.com/29683145/132707413-4932e6cc-f3b4-4801-a9aa-bd6523a9fba5.png) Problem Solved ![image](https://user-images.githubusercontent.com/29683145/132707718-7615519c-b5c3-4dd4-8e26-3fe5b7158f58.png)

@akashchgupta If you're having the same output, that's what I'm asking