MCPcopy Index your code
hub / github.com/tensorpack/tensorpack / devices

Method devices

tensorpack/utils/nvml.py:178–183  ·  view source on GitHub ↗

Returns: [NvidiaDevice]: a list of devices

(self)

Source from the content-addressed store, hash-verified

176 return c_count.value
177
178 def devices(self):
179 """
180 Returns:
181 [NvidiaDevice]: a list of devices
182 """
183 return [self.device(i) for i in range(self.num_devices())]
184
185 def device(self, idx):
186 """Get a specific GPU device

Callers 2

nvml.pyFile · 0.80
collect_env_infoFunction · 0.80

Calls 2

deviceMethod · 0.95
num_devicesMethod · 0.95

Tested by

no test coverage detected