fix clipped ppo

2025-12-17 11:10:20 +01:00 · 2018-02-16 13:22:10 -05:00
parent 85afb86893
commit 39a28aba95
7 changed files with 51 additions and 39 deletions
--- a/utils.py
+++ b/utils.py
@@ -132,6 +132,7 @@ def parse_int(value):

 def set_gpu(gpu_id):
    os.environ['CUDA_VISIBLE_DEVICES'] = str(gpu_id)
+    os.environ['NVIDIA_VISIBLE_DEVICES'] = str(gpu_id)


 def set_cpu():