()
| 5 | import subprocess |
| 6 | |
| 7 | def getAcceleratorName(): |
| 8 | try: |
| 9 | deviceName = subprocess.check_output(['nvidia-smi', '--query-gpu=name', '--format=csv,noheader']) |
| 10 | return deviceName.decode('utf-8').strip() |
| 11 | except FileNotFoundError: |
| 12 | return("nvidia-smi not found.") |
| 13 | |
| 14 | def isGPU(): |
| 15 | return os.path.isfile('/proc/driver/nvidia/version') |