Turn setup_ci_environment into command
Summary: Pull Request resolved: https://github.com/pytorch/pytorch/pull/26163
Test Plan: Imported from OSS
Differential Revision: D17366536
Pulled By: pietern
fbshipit-source-id: 07181a77aaeba5457aa716ceac9cc404aacefe5f