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

在python3.8和pytorch1.8.1的基础上安装tensorflow

以往看的模型都是用pytorch跑就行,但是看到的一个创新点是用tensorflow跑的,迫于无奈,如果想要融入进去就得安装TensorFlow…

  1. 根据官网的指示,使用 tensorflow2 的 python 版本应当大于 3.5
  2. 如果使用 python 3.8,那么 Tensorflow 版本要大于 2.2

最大的感悟是,正确是安装思路应该是:

  1. 了解阶段:设备情况----> 能接受的最大cuda/cudnn版本---->pytorch/tensorflow版本。
  2. 版本选择阶段:先确定好想安装的pytorch/tensorflow版本---->确定其他配件的版本
  3. 最后再从底层文件依次安装。

每次根据电脑性能选择了,接受范围内最适配的底层包进行安装,最后都会发现,我想要安装的最终文件要求更高。又是苟且安装成功的一天…

一、下载cudnn

一定要选好恰当的版本。
版本选择可以参考:tensorflow与cuda版本对应
下载网址:https://developer.nvidia.com/rdp/cudnn-archive(需要注册 NVIDIA 开发者账号并加入开发者计划(免费))
在这里插入图片描述
在这里插入图片描述

二、 解压+移动

解压结果如下:
在这里插入图片描述
1.打开cuda的安装目录(我的版本是11.1):C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v11.1
2.将解压出来的三个文件夹复制到cuda
在这里插入图片描述
此处利用管理者权限选择**“替换”**

三、安装tensorflow-gpu==2.4.0

pip install tensorflow-gpu==2.4.0 -i https://mirrors.aliyun.com/pypi/simple  # 阿里云镜像

需要注意的是,一定要写tensorflow-gpu==2.4.0而不是 “tensorflow==2.4.0”

四、验证

接下来验证tensorflow是否安装成功。

python # 输入python进入代码编写界面import tensorflow as tf  # 导入库
tf.__version__  # 查看版本号

在这里插入图片描述
最后测试GPU情况

tf.test.is_gpu_available()

在这里插入图片描述
然鹅没成功,返回“False”了。
解决办法参考:TensorFlow2.4 GPU版安装教程
观察发现报错原因:缺少了某些 CUDA 库文件(尤其是 cusolver64_10.dll)
在这里插入图片描述
打开cuda安装目录下的bin文件,即C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v11.1\bin
将**“cusolver64_11.dll” 文件复制重命名为cusolver64_10.dll**即可
在这里插入图片描述
再次运行测试代码,返回True。
在这里插入图片描述
小插曲:
tensorflow安装成功了,但是numpy和pandas、matplotlib之间的版本依赖关系出现了错误。
在这里插入图片描述
模拟一下安装过程检查版本依赖,

pip install --dry-run matplotlib==3.3.4 pandas==1.2.5 multidict==5.2.0  # 模拟一下安装过程检查版本依赖

没有报错就直接安装了。


安装pydoe,直接安装即可

pip install pyDOE -i https://mirrors.aliyun.com/pypi/simple

参考博客

1.Anaconda同时安装PyTorch和TensorFlow2.x版本教程(Windows + GPU/CUDA)
2.windows10下CUDA11.1、CUDNN11.1、tensorflow-gpu 2.4.1和torch安装教程
3. win 10 下安装深度学习环境(tensorflow + pytorch, CUDA + CUDNN)

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

相关文章:

  • Flink是如何实现物理分区?
  • 深入解析C++原子指针std::atomic
  • GIS地理信息系统建设:高精度3D建模
  • [Linux入门] 初学者入门:Linux DNS 域名解析服务详解
  • Anaconda常用命令及环境管理指南
  • kali [DNS劫持] 实验(详细步骤)
  • Containerd简介
  • 【Ubuntu】发展历程
  • 如何做数据增强?
  • 框架式3D打印机结构设计cad【9张】三维图+设计说明书
  • 【升级U8+】不能将值 NULL 插入列 ‘LocaleID‘,表 ‘hr_sys_function_Base‘;列不允许有 Null 值。
  • flink查看taskManager日志
  • 数组内存学习
  • Django5.1(130)—— 表单 API一(API参考)
  • 【Luogu】每日一题——Day14. P5960 【模板】差分约束 (图论 + 最短路)
  • Nacos-服务注册,服务发现(二)
  • mac版SVN客户端
  • Pythong高级入门Day5
  • npm ERR! cb() never called!
  • 昇思学习营-DeepSeek-R1-Distill-Qwen-1.5B 模型LoRA微调学习心得
  • 数据结构 二叉树(2)---二叉树的实现
  • 2025年SEVC SCI2区,混沌编码量子粒子群算法QPSO+柔性车间调度,深度解析+性能实测
  • 每日一题7.26
  • WorkManager vs Flow 适用场景分析
  • 抖音短视频矩阵系统源码搭建---底层框架5年开发分享
  • idea中无法删除模块,只能remove?
  • 二叉搜索树(Binary Search Tree)详解与java实现
  • 代码随想录打卡第十五天
  • 睡眠函数 Sleep() C语言
  • AI Agent开发学习系列 - langchain: 本地大模型调用