File tree Expand file tree Collapse file tree 1 file changed +3
-3
lines changed Expand file tree Collapse file tree 1 file changed +3
-3
lines changed Original file line number Diff line number Diff line change @@ -52,12 +52,12 @@ Utils
52
52
grab_gpus
53
53
~~~~~~~~~
54
54
55
- You can call the `grab_gpus(num_gpus, gpu_select) ` function to check the
55
+ You can call the :code: `grab_gpus(num_gpus, gpu_select) ` function to check the
56
56
available gpus and set the `CUDA_VISIBLE_DEVICES ` environment variable as need
57
57
be. It determines if a GPU is available by checking if the memory-usage is 0%.
58
58
59
59
I have found this useful as I have a shared gpu server and like to use
60
- tensorflow which is very greedy and calls to `tf.Session() ` grabs all available gpus.
60
+ tensorflow which is very greedy and calls to :code: `tf.Session() ` grabs all available gpus.
61
61
62
62
E.g.
63
63
@@ -80,7 +80,7 @@ option is not necessary, and it only serves to restrict the search space for
80
80
the grab_gpus.
81
81
82
82
You can adjust the memory threshold for determining if a GPU is free/used with
83
- the `gpu_fraction ` parameter (default is 1):
83
+ the :code: `gpu_fraction ` parameter (default is 1):
84
84
85
85
.. code :: python
86
86
You can’t perform that action at this time.
0 commit comments