Code
Hub
Workspaces
Following
Trending
Connect
MCP
copy
Create free account
hub
/
github.com/apache/singa
/ get_gpu_ids
Function
get_gpu_ids
python/singa/device.py:34–36 ·
view source on GitHub ↗
()
Source
from the content-addressed store, hash-verified
32
33
34
def
get_gpu_ids():
35
assert singa.USE_CUDA,
'SINGA has not been compiled with CUDA enabled.'
36
return
singa.Platform.GetGPUIDs()
37
38
39
def
get_gpu_mem_size(id):
Callers
nothing calls this directly
Calls
1
GetGPUIDs
Method · 0.80
Tested by
no test coverage detected