欢迎来到尧图网

客户服务 关于我们

您的位置:首页 > 新闻 > 国际 > 安装 tensorflow-2.10.0 支持 gpu

安装 tensorflow-2.10.0 支持 gpu

2025/5/24 21:09:52 来源:https://blog.csdn.net/waterHBO/article/details/148153009  浏览:    关键词:安装 tensorflow-2.10.0 支持 gpu

1. 起因, 目的:

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

2. 先看效果

无。

3. 过程:

代码
import platform
import sys
import os
import subprocess
import shutildef 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 = Falsefor path in possible_paths:if os.path.exists(path):print(f"已检测到 CUDA Toolkit 安装路径: {path}")found = Truebreakif 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
系统架构: AMD64Python 版本: 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"""

4. 结论 + todo

  • 繁琐是难免的。容易吗,不容易!

希望对大家有帮助。

版权声明:

本网仅为发布的内容提供存储空间,不对发表、转载的内容提供任何形式的保证。凡本网注明“来源:XXX网络”的作品,均转载自其它媒体,著作权归作者所有,商业转载请联系作者获得授权,非商业转载请注明出处。

我们尊重并感谢每一位作者,均已注明文章来源和作者。如因作品内容、版权或其它问题,请及时与我们联系,联系邮箱:809451989@qq.com,投稿邮箱:809451989@qq.com

热搜词