安装 tensorflow-2.10.0 支持 gpu-CSDN博客

本文链接：https://blog.csdn.net/waterHBO/article/details/148153009

1. 起因，目的:

重新安装 tensorflow gpu，因为之前的项目都删掉了。遇到一点问题，记录一下。
来回折腾，本质是我的 python3.11 版本太高了。换成 python3.9 就行了。
坚决不使用 AI 扩写。垃圾而恶心。

2. 先看效果

无。

3. 过程:

代码

import platform
import sys
import os
import subprocess
import shutil


def check_os():
    os_type = platform.system()
    arch = platform.machine()
    print(f"操作系统: {os_type}")
    print(f"系统架构: {arch}")


def check_python():
    print(f"Python 版本: {platform.python_version()}")
    print(f"Python 位数: {platform.architecture()[0]}")


def check_cuda_toolkit():
    possible_paths = [
        "C:\\Program Files\\NVIDIA GPU Computing Toolkit\\CUDA",
        "/usr/local/cuda",
    ]
    found = False
    for path in possible_paths:
        if os.path.exists(path):
            print(f"已检测到 CUDA Toolkit 安装路径: {path}")
            found = True
            break
    if not found:
        print("未检测到 CUDA Toolkit 安装路径")


def check_nvidia_smi():
    try:
        result = subprocess.run(["nvidia-smi"], capture_output=True, text=True)
        if result.returncode == 0:
            print("✅ 已检测到 NVIDIA GPU：")
            print(result.stdout.split("\n")[0])  # 打印首行信息
        else:
            print("⚠️ 未检测到 NVIDIA GPU 或未安装驱动")
    except FileNotFoundError:
        print("⚠️ 未安装 NVIDIA 驱动（nvidia-smi 命令不可用）")


def main():
    print("=== 系统信息检测 ===")
    check_os()
    print()
    check_python()
    print()
    check_cuda_toolkit()
    print()
    check_nvidia_smi()
    print("====================")


if __name__ == "__main__":
    main()

"""
输出：

=== 系统信息检测 ===
操作系统: Windows
系统架构: AMD64

Python 版本: 3.11.1
Python 位数: 64bit

已检测到 CUDA Toolkit 安装路径: C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA

✅ 已检测到 NVIDIA GPU：
Fri May 23 05:02:23 2025       
====================


而且：

(venv) C:\Users\Administrator\Code\python_nlp>nvcc --version
nvcc: NVIDIA (R) Cuda compiler driver                  
Copyright (c) 2005-2022 NVIDIA Corporation             
Built on Tue_Mar__8_18:36:24_Pacific_Standard_Time_2022
Cuda compilation tools, release 11.6, V11.6.124        
Build cuda_11.6.r11.6/compiler.31057947_0              

(venv) C:\Users\Administrator\Code\python_nlp>         

(venv) C:\Users\Administrator\Code\python_nlp>nvidia-smi
Fri May 23 05:07:25 2025                                                                   
+-----------------------------------------------------------------------------------------+
| NVIDIA-SMI 572.70                 Driver Version: 572.70         CUDA Version: 12.8     |
|-----------------------------------------+------------------------+----------------------+
| GPU  Name                  Driver-Model | Bus-Id          Disp.A | Volatile Uncorr. ECC |
| Fan  Temp   Perf          Pwr:Usage/Cap |           Memory-Usage | GPU-Util  Compute M. |
|                                         |                        |               MIG M. |
|  GPU   GI   CI              PID   Type   Process name                        GPU Memory |
|        ID   ID                                                               Usage      |
|=========================================================================================|
|    0   N/A  N/A            2300    C+G   ...ram Files\Tencent\QQNT\QQ.exe      N/A      |
|    0   N/A  N/A            8288    C+G   ...h_cw5n1h2txyewy\SearchApp.exe      N/A      |
|    0   N/A  N/A            9224    C+G   ...xyewy\ShellExperienceHost.exe      N/A      |
|    0   N/A  N/A           10644    C+G   C:\Windows\explorer.exe               N/A      |
|    0   N/A  N/A           11168    C+G   ...1.1.1\jbr\bin\jcef_helper.exe      N/A      |
|    0   N/A  N/A           12428    C+G   ...5n1h2txyewy\TextInputHost.exe      N/A      |
|    0   N/A  N/A           13452    C+G   ...Chrome\Application\chrome.exe      N/A      |
|    0   N/A  N/A           14156    C+G   ...acted\runtime\WeChatAppEx.exe      N/A      |
|    0   N/A  N/A           14396    C+G   ...Chrome\Application\chrome.exe      N/A      |
|    0   N/A  N/A           16664    C+G   ....0.3240.76\msedgewebview2.exe      N/A      |
+-----------------------------------------------------------------------------------------+

(venv) C:\Users\Administrator\Code\python_nlp>
(venv) C:\Users\Administrator\Code\python_nlp>



请根据我的系统信息，帮我配置 tensorflow gpu , 如何安装，请给出具体步骤。

# 我的 python3.11， 版本太高了， 不行。换成 python3.9 可以的！
pip install tensorflow==2.10.0 -i https://pypi.org/simple

"""