From 845195acec0763a9e255f246b097f4f6399f53ed Mon Sep 17 00:00:00 2001 From: Michael Mykhaylov Date: Wed, 25 Oct 2023 10:59:12 -0700 Subject: [PATCH] Replace deprecated CUDA check --- intermediate_source/reinforcement_ppo.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/intermediate_source/reinforcement_ppo.py b/intermediate_source/reinforcement_ppo.py index 6501e98971e..1b75e21e51a 100644 --- a/intermediate_source/reinforcement_ppo.py +++ b/intermediate_source/reinforcement_ppo.py @@ -137,7 +137,7 @@ # actually return ``frame_skip`` frames). # -device = "cpu" if not torch.has_cuda else "cuda:0" +device = "cpu" if not torch.cuda.is_available() else "cuda:0" num_cells = 256 # number of cells in each layer i.e. output dim. lr = 3e-4 max_grad_norm = 1.0