Fix random seed problem

Author: ikostrikov2

README.md

@@ -20,7 +20,7 @@ Contributions are very welcome. If you know how to make this code better, don't
 ## Usage
 ```bash
 # Works only wih Python 3.
-OMP_NUM_THREADS=1 python3 main.py --env-name "PongDeterministic-v4" --num-processes 16
+python3 main.py --env-name "PongDeterministic-v4" --num-processes 16
 ```
 
 This code runs evaluation in a separate thread in addition to 16 processes.

main.py

@@ -44,11 +44,11 @@
 
 if __name__ == '__main__':
     os.environ['OMP_NUM_THREADS'] = '1'
+    os.environ['CUDA_VISIBLE_DEVICES'] = ""
 
     args = parser.parse_args()
 
-    # uncomment when it's fixed in pytorch
-    # torch.manual_seed(args.seed)
+    torch.manual_seed(args.seed)
     env = create_atari_env(args.env_name)
     shared_model = ActorCritic(
         env.observation_space.shape[0], env.action_space)