From 718ce573acca12922963a60767861f0f2f5a28aa Mon Sep 17 00:00:00 2001 From: cryptexis Date: Tue, 20 Feb 2024 11:13:20 -0800 Subject: [PATCH] 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: https://github.com/facebookresearch/Pearl/pull/66 Reviewed By: jb3618columbia Differential Revision: D53948188 Pulled By: rodrigodesalvobraz fbshipit-source-id: 460ad0dcebcbd593ec4fb13f2b25960983c20bab --- pearl/utils/scripts/benchmark_offline_rl.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/pearl/utils/scripts/benchmark_offline_rl.py b/pearl/utils/scripts/benchmark_offline_rl.py index 13e6285f..a05439a6 100644 --- a/pearl/utils/scripts/benchmark_offline_rl.py +++ b/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)