当前位置: 首页 > news >正文

5202年安装TensorFlow纪实

没想到5202年了,还有安装TensorFlow的需求,N年不用,TensorFlow还是那么难用,正确安装依旧困难。

Windows

根据 Tensorflow、Pytorch与Python、CUDA版本的对应关系(更新时间:2025年4月),windows的官方tensorflow-gpu最高仅支持2.10.0,因此我们就装这个版本。

首先,用pip安装:

pip install tensorflow-gpu==2.10.0

这样安装之后,TensorFlow大概率无法使用显卡,而是会报一串dll not found的错。对此,一种解决方法是将对应的dll文件复制到C:/Windows/System路径下,但有时,我们系统的cuda和cudnn版本并不是TensorFlow需要的版本,因此可以在conda环境中安装对应的cuda和cudnn包。

为此,首先可以到路径C:\Users\XX\miniforge3\envs\xx\Lib\site-packages\tensorflow\python\platform下的build_info.py中查看该TensorFlow版本的编译信息,例如:

build_info = collections.OrderedDict([('cpu_compiler', 'C:/Program Files (x86)/Microsoft Visual Studio/2019/Community/VC/Tools/MSVC/14.29.30133/bin/HostX64/x64/cl.exe'), ('cuda_compute_capabilities', ['sm_35', 'sm_50', 'sm_60', 'sm_70', 'sm_75', 'compute_80']), ('cuda_version', '64_112'), ('cudart_dll_name', 'cudart64_112.dll'), ('cudnn_dll_name', 'cudnn64_8.dll'), ('cudnn_version', '64_8'), ('is_cuda_build', True), ('is_rocm_build', False), ('is_tensorrt_build', False), ('msvcp_dll_names', 'msvcp140.dll,msvcp140_1.dll'), ('nvcuda_dll_name', 'nvcuda.dll')])

由此可知,该版本的TensorFlow是在cuda11.2、cudnn8的环境下编译的,因此,我们在conda环境中继续安装:

conda install cudatoolkit=11.2 cudnn=8

这样TensorFlow就能正常识别GPU了。


http://www.dtcms.com/a/270577.html

相关文章:

  • 【LeetCode207.课程表】以及变式
  • 暑假算法日记第五天
  • [2025CVPR]Mr. DETR:检测Transformer的多路由指导训练解析
  • Mysql组合索引的update在多种情况下的间隙锁的范围(简单来说)
  • 141-CEEMDAN-VMD-Transformer-BiLSTM-ABKDE多变量区间预测模型!
  • [数学基础] 矩阵的秩及其应用
  • El-Select组件实现模糊查询与失焦赋值
  • 第6章应用题
  • 学术绘图(各种神经网络)
  • 5.注册中心横向对比:Nacos vs Eureka vs Consul —— 深度解析与科学选型指南
  • Microsoft AZ-305 Exam Question
  • Flutter基础(前端教程⑦-Http和卡片)
  • Flutter基础(前端教程⑥-按钮切换)
  • 《重构项目》基于Apollo架构设计的项目重构方案(多种地图、多阶段、多任务、状态机管理)
  • 【教程】在ubuntu安装Edge浏览器
  • 工业通讯网关在电子制造中的核心作用——从DeviceNet到Modbus TCP的智能转换
  • 家庭网络中的服务器怎么对外提供服务?
  • 跨平台ROS2视觉数据流:服务器运行IsaacSim+Foxglove本地可视化全攻略
  • 【网络】Linux 内核优化实战 - net.ipv4.tcp_dsack
  • NHibernate案例
  • scp:上传大型数据集到实验室服务器
  • 2025年人工智能、虚拟现实与交互设计国际学术会议
  • 蛋白质序列-kappa参数计算算法解读
  • Linux——I/O复用
  • django 一个表中包括id和parentid,如何通过parentid找到全部父爷id
  • 微信小程序91~100
  • 什么是高防 IP?从技术原理到实战部署的深度解析
  • 制作MikTex本地包可用于离线安装包
  • springBoot使用XWPFDocument 和 LoopRowTableRenderPolicy 两种方式填充数据到word模版中
  • ARM单片机滴答定时器理解与应用(一)(详细解析)