-
Notifications
You must be signed in to change notification settings - Fork 0
/
Copy pathGet_Supported_GPUs.py
61 lines (50 loc) · 1.83 KB
/
Get_Supported_GPUs.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
#Get_Supported_GPUs
from __future__ import print_function
from pyVim.connect import SmartConnect, Disconnect
from pyVmomi import vim
from pyVmomi import vmodl
import argparse
import atexit
import getpass
import ssl
from pyVmomi.VmomiSupport import NoneType
def main():
context = None
if hasattr(ssl, '_create_unverified_context'):
context = ssl._create_unverified_context()
si = SmartConnect(host="vsa01.wondernerd.local",
user="[email protected]",
pwd="VMware-1",
port=443,
sslContext=context)
if not si:
print("Could not connect to the specified host using specified "
"username and password")
return -1
atexit.register(Disconnect, si)
###################################################
# New Content
###################################################
HostContent=si.content
TempHold = HostContent.viewManager.CreateContainerView(
HostContent.rootFolder,[vim.HostSystem], True)
for managed_object_ref in TempHold.view:
print(managed_object_ref.name)
try:
#print(isinstance(managed_object_ref.config, NoneType))
if isinstance(managed_object_ref.config, NoneType) == False:
if managed_object_ref.config.sharedPassthruGpuTypes != []:
for GPU_Config in managed_object_ref.config.graphicsInfo:
print(GPU_Config.deviceName)
else:
print("No GPUs found")
else:
print("Host powered off")
except:
print("Host not accessable")
###################################################
# End New Content
###################################################
return 0
if __name__ == "__main__":
main()