[RLlib] Fix/torch gpu test (ray-project#60246)

Signed-off-by: peterxcli <peterxcli@gmail.com>

Author: HassamSheikh

rllib/BUILD.bazel

@@ -2283,15 +2283,20 @@ py_test(
     deps = [":conftest"],
 )
 
-# TODO(Artur): Comment this back in as soon as we can test with GPU
-# py_test(
-#    name = "test_torch_rl_module_gpu",
-#    main = "core/rl_module/torch/tests/test_torch_rl_module.py",
-#    tags = ["team:rllib", "core", "gpu", "exclusive"],
-#    size = "medium",
-#    srcs = ["core/rl_module/torch/tests/test_torch_rl_module.py"],
-#    args = ["TestRLModuleGPU"],
-# )
+py_test(
+    name = "test_torch_rl_module_gpu",
+    size = "medium",
+    srcs = ["core/rl_module/torch/tests/test_torch_rl_module.py"],
+    args = ["TestRLModuleGPU"],
+    main = "core/rl_module/torch/tests/test_torch_rl_module.py",
+    tags = [
+        "core",
+        "exclusive",
+        "gpu",
+        "team:rllib",
+    ],
+    deps = [":conftest"],
+)
 
 py_test(
     name = "test_multi_rl_module",