Replace deprecated CUDA check (#2621)

mikemykhaylov · svekars · web-flow · commit 62468fa612da · 2023-10-25T13:10:35.000-07:00
Co-authored-by: Svetlana Karslioglu &lt;svekars@meta.com&gt;
diff --git 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

Original file line number	Diff line number	Diff line change
`@@ -137,7 +137,7 @@`
`137`	`137`	# actually return ``frame_skip`` frames).
`138`	`138`	`#`
`139`	`139`
`140`		`-device = "cpu" if not torch.has_cuda else "cuda:0"`
	`140`	`+device = "cpu" if not torch.cuda.is_available() else "cuda:0"`
`141`	`141`	`num_cells = 256 # number of cells in each layer i.e. output dim.`
`142`	`142`	`lr = 3e-4`
`143`	`143`	`max_grad_norm = 1.0`