python 查看 nvida 驱动、 cuda、pytorch、tensorflow的版本

nvidia 显卡、驱动、cuda 的版本

import subprocess
command = '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 2022       
+-----------------------------------------------------------------------------+
| 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    P8     6W / 120W |   1172MiB /  6144MiB |      8%      Default |
|                               |                      |                  N/A |
+-------------------------------+----------------------+----------------------+
                                                                               
+-----------------------------------------------------------------------------+
| Processes:                                                                  |
|  GPU   GI   CI        PID   Type   Process name                  GPU Memory |
|        ID   ID                                                   Usage      |
|=============================================================================|
|    0   N/A  N/A      1288    C+G   Insufficient Permissions        N/A      |
|    0   N/A  N/A      1856    C+G   ...icrosoft VS Code\Code.exe    N/A      |
|    0   N/A  N/A      2100    C+G   ...4__8wekyb3d8bbwe\Todo.exe    N/A      |
|    0   N/A  N/A      3540    C+G   ...y\ShellExperienceHost.exe    N/A      |
|    0   N/A  N/A      3700    C+G   C:\Windows\explorer.exe         N/A      |
|    0   N/A  N/A      4324    C+G   ...lPanel\SystemSettings.exe    N/A      |
|    0   N/A  N/A      6416    C+G   ...artMenuExperienceHost.exe    N/A      |
|    0   N/A  N/A      8504    C+G   ...ekyb3d8bbwe\YourPhone.exe    N/A      |
|    0   N/A  N/A      9584    C+G   ...me\Application\chrome.exe    N/A      |
|    0   N/A  N/A     10384    C+G   ...kyb3d8bbwe\Calculator.exe    N/A      |
|    0   N/A  N/A     10492    C+G   ...5n1h2txyewy\SearchApp.exe    N/A      |
|    0   N/A  N/A     10708    C+G   ...8\extracted\WeChatApp.exe    N/A      |
|    0   N/A  N/A     11576    C+G   ...cw5n1h2txyewy\LockApp.exe    N/A      |
|    0   N/A  N/A     11668    C+G   ...ge\Application\msedge.exe    N/A      |
|    0   N/A  N/A     12372    C+G   ...2txyewy\TextInputHost.exe    N/A      |
|    0   N/A  N/A     14428    C+G   ...wekyb3d8bbwe\Video.UI.exe    N/A      |
|    0   N/A  N/A     14872    C+G   ...tracted\WechatBrowser.exe    N/A      |
|    0   N/A  N/A     15020    C+G   ...m Files\Typora\Typora.exe    N/A      |
+-----------------------------------------------------------------------------+

pytorch 的版本

import torch

print("pytorch的版本：", torch.__version__)
print("pytorch是否使用gpu：", torch.cuda.is_available())

pytorch的版本： 1.10.2+cpu
pytorch是否使用gpu： False

tensorflow 的版本

import tensorflow

print("tensorflow的版本：", tensorflow.__version__)

tensorflow的版本： 2.7.0