nvidia 显卡、驱动、cuda 的版本
import subprocesscommand = 'nvidia-smi' # 终端命令open_process = subprocess.Popen(command, # 终端命令stdout=subprocess.PIPE,shell=True # 允许执行shell程序)cmd_out = open_process.stdout.read() # 读取信息,并返回值open_process.stdout.close() # 关闭输出管道print("终端输出:", cmd_out.decode(encoding="gbk"))
终端输出: Thu Feb 24 16:38:45 +-----------------------------------------------------------------------------+| NVIDIA-SMI 456.81 Driver Version: 456.81 CUDA Version: 11.1||-------------------------------+----------------------+----------------------+| GPU Name TCC/WDDM | Bus-Id Disp.A | Volatile Uncorr. ECC || Fan Temp Perf Pwr:Usage/Cap| Memory-Usage | GPU-Util Compute M. || | |MIG M. ||===============================+======================+======================|| 0 GeForce GTX 166... WDDM | 00000000:01:00.0 On | N/A || 31% 35C P86W / 120W | 1172MiB / 6144MiB |8%Default || | | N/A |+-------------------------------+----------------------+----------------------++-----------------------------------------------------------------------------+| Processes:|| GPU GI CI PID Type Process name GPU Memory || ID ID Usage||=============================================================================|| 0 N/A N/A1288 C+G Insufficient Permissions N/A|| 0 N/A N/A1856 C+G ...icrosoft VS Code\Code.exe N/A|| 0 N/A N/A2100 C+G ...4__8wekyb3d8bbwe\Todo.exe N/A|| 0 N/A N/A3540 C+G ...y\ShellExperienceHost.exe N/A|| 0 N/A N/A3700 C+G C:\Windows\explorer.exe N/A|| 0 N/A N/A4324 C+G ...lPanel\SystemSettings.exe N/A|| 0 N/A N/A6416 C+G ...artMenuExperienceHost.exe N/A|| 0 N/A N/A8504 C+G ...ekyb3d8bbwe\YourPhone.exe N/A|| 0 N/A N/A9584 C+G ...me\Application\chrome.exe N/A|| 0 N/A N/A10384 C+G ...kyb3d8bbwe\Calculator.exe N/A|| 0 N/A N/A10492 C+G ...5n1h2txyewy\SearchApp.exe N/A|| 0 N/A N/A10708 C+G ...8\extracted\WeChatApp.exe N/A|| 0 N/A N/A11576 C+G ...cw5n1h2txyewy\LockApp.exe N/A|| 0 N/A N/A11668 C+G ...ge\Application\msedge.exe N/A|| 0 N/A N/A12372 C+G ...2txyewy\TextInputHost.exe N/A|| 0 N/A N/A14428 C+G ...wekyb3d8bbwe\Video.UI.exe N/A|| 0 N/A N/A14872 C+G ...tracted\WechatBrowser.exe N/A|| 0 N/A N/A15020 C+G ...m Files\Typora\Typora.exe N/A|+-----------------------------------------------------------------------------+
pytorch 的版本
import torchprint("pytorch的版本:", torch.__version__)print("pytorch是否使用gpu:", torch.cuda.is_available())
pytorch的版本: 1.10.2+cpupytorch是否使用gpu: False
tensorflow 的版本
import tensorflowprint("tensorflow的版本:", tensorflow.__version__)
tensorflow的版本: 2.7.0