Revert "refactor the gpu setup on more time"

This reverts commit 6074ffd772909c3fc1c3e1a825d319b20338e594.