wip: fixing device id (#66)

Summary:
High, many people do not have the second CUDA device. Personally spent a lot of time debugging cuda error and trying to understand where the error comes from. :)

Pull Request resolved: #66

Reviewed By: jb3618columbia

Differential Revision: D53948188

Pulled By: rodrigodesalvobraz

fbshipit-source-id: 460ad0dcebcbd593ec4fb13f2b25960983c20bab

pearl/utils/scripts/benchmark_offline_rl.py

@@ -241,7 +241,7 @@ def evaluate_offline_rl(
 
 
 if __name__ == "__main__":
-    device_id = 1  # use -1 for CPU, 0, 1, for cuda
+    device_id = 0  # use -1 for CPU, 0, 1, for cuda
     experiment_seed = 100
     env_name = "HalfCheetah-v4"
     env = GymEnvironment(env_name)