友情提醒:所有蓝字均为超链接

目录

  • 1 本机环境及碎碎念
  • 2 排坑指南
    • 2.1 TensorFlow及CUDA、cuDNN版本
    • 2.2 CUDA版本选择
    • 2.2 cuDNN版本选择
  • 3 安装教程
    • 3.1 CUDA安装
    • 3.2 cuDNN安装
    • 3.3 TensorFlow-gpu 2.5安装
  • 参考

1 本机环境及碎碎念

联想拯救者R7000p 2021,GPU:RTX3050Ti,系统:win10

30系显卡由于刚出不久,版本驱动、CUDA版本都很新,之前的网上的很多博客都是针对旧版本的,并且也没有对版本有个很好的汇总,给新手安装带来一定困难。博主刚好换了新电脑(苦逼实验室没服务器,抢了入门的R7000P),在安装过程中排了一些坑,也对版本有了更新的认识,希望能总结一下,给大家减少不必要的麻烦。不过,博主也是小白,如有不对或者不全面的地方希望大家能在评论区指出,一起交流哈!

2 排坑指南

2.1 TensorFlow及CUDA、cuDNN版本

对于安装过GPU版本的小伙伴们都知道,不同版本的tensorflow对应着不同的cuda以及cudnn版本,具体可以从TensorFlow手册上查找,我截图放在下面:

2.2 CUDA版本选择

然而不同的CUDA是对显卡的驱动有要求的,这里给出NVIDIA官方给出的对应表,所以我们需要根据自己本机的GPU选用对应的CUDA版本。

首先,在桌面【右键】-打开【NVIDIA 控制面板】-选择【帮助】-【系统信息】,再打开的系统信息中选择【组件】,即可看到本机GPU对应支持的CUDA版本:

或者可以在终端输入命令nvidia-smi查看GPU驱动版本:

综上所述,对应着表来看,我们所需的CUDA版本就是11.2.1无疑了,CUDA下载可以通过NVIDIA官网下载,这里我把cuda及cudnn的包下好放在百度云上了,大家自取:

链接:https://pan.baidu.com/s/1piO6WL7KfbMc-6FrgH9X1g
提取码:1024

2.2 cuDNN版本选择

可以登陆NVIDIA官网进行查看,下载cuDNN还要求注册,需要捣鼓一阵子,界面和下头一样:

这里根据CUDA版本提示选择8.1.0即可,不过上头的8.1.1应该也可以,如有不对,评论区请指出哈。和上面一样,百度云链接如下:

链接:https://pan.baidu.com/s/1piO6WL7KfbMc-6FrgH9X1g
提取码:1024

3 安装教程

安装主要包括CUDA、cuDNN以及TensorFlow2.5的安装,下面一步一步来

3.1 CUDA安装

双击cuda_11.2.1_461.09_win10.exe,进入安装程序:

这里是临时选择临时提取的文件夹,不用管直接ok:

提取完毕,程序会检查兼容性:

选择【同意并继续】,选择【自定义】安装:

在CUDA中取消勾选Visual Studio Integration,这是VS的插件,容易导致安装失败,我因为用pycharm所以就不勾选了

接着下一步,选择安装位置,我直接默认了,就进入安装准备直接安装即可。

等待安装完毕,可以检查是否安装成功,在终端键入命令nvcc -V,查看CUDA版本,能显示则成功了。

3.2 cuDNN安装

cuDNN大家也看到了,就是个压缩包,大家先解压,如图所示:

然后复制到CUDA的安装路径即可,我的是C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v11.2

最后,跟着我配置下系统环境即可:
右键【此电脑】又名计算机—【属性】—【高级系统设置】-【环境变量】-下半部分【系统变量】里找到【Path】,双击打开(或者点编辑)

点击【新建】,补全如下环境变量,如果你之前安装成功了CUDA,2、3条环境应该是系统已经自动添加了的~

至此,你的CUDA及cuDNN已经安装好了

3.3 TensorFlow-gpu 2.5安装

  1. 安装anaconda
    这个不用多说了,进入anaconda官网下载页面,直接下载即可,具体安装步骤不懂的可以查看其他帖子,注意勾选添加系统环境就行。
  2. 新建虚拟环境
    建议新建个环境来捣鼓,在终端键入命令
conda create -n tf25 python=3.7   # 这里的tf25就是你的虚拟环境名


会提示你安装所需包,选择y即可,安装完成后输入conda activate tf25激活进入环境。

注意这里括号前后的变化,说明进入了虚拟环境

  1. 安装TensorFlow2.5
    这里补充下为什么安装2.5版本,我是买电脑之前一直使用Google的Colab的GPU训练平台,上面用的TensorFlow版本就是2.5,兼容性也是很不错的,并且回到第一节的版本表,CUDA11.2应该也只支持2.4和2.5版本,所以一步到位吧~
    在刚刚进入的环境里键入命令pip install tensorflow-gpu==2.5,如下图所示开始安装

    注意事项:红框内代表我已更换了pip国内源,一行命令即可:
pip config set global.index-url https://pypi.tuna.tsinghua.edu.cn/simple
以下是国内的主流镜像阿里云 http://mirrors.aliyun.com/pypi/simple/ 中国科技大学 https://pypi.mirrors.ustc.edu.cn/simple/ 豆瓣(douban) http://pypi.douban.com/simple/ 清华大学 https://pypi.tuna.tsinghua.edu.cn/simple/ 中国科学技术大学 http://pypi.mirrors.ustc.edu.cn/simple/

由于之前我们已经安装过CUDA和cuDNN,所以这里安装好TensorFlow后直接可以进入测试。

首先,可以输入conda list查看当前环境下所安装的包:

其次,输入python,进入python终端,分别输入一下命令:

import tensorflow as tf
print(tf.__version__)
print('GPU', tf.test.is_gpu_available())

分别打印出 2.5.0GPU True,即代表环境安装成功,去开启你们“快乐”的调参之旅吧!

print(tf.__version__)
2.5.0
print('GPU', tf.test.is_gpu_available())
GPU True

参考

[1]: https://blog.csdn.net/weixin_43233764/article/details/111536819
[2]: https://www.bilibili.com/read/cv10662122
[3]: https://blog.csdn.net/weixin_41194129/article/details/114238772
[4]: https://docs.nvidia.com/cuda/cuda-toolkit-release-notes/index.html
[5]: https://blog.csdn.net/mayyyyyya/article/details/113656914
[6]: https://blog.csdn.net/m0_37412775/article/details/109807085

【踩坑教程】win10环境下RTX3050Ti安装Tensorflow-gpu2.5+CUDA11.2.1+cudnn8.1.0相关推荐

  1. win10环境下下载安装openpose(only cpu)并在pycharm中运行代码(超详细)

    win10环境下下载安装openpose(only cpu)并在pycharm中运行代码(超详细) (一)前言 (二)准备工作 (三)友情提醒 (四)详细安装步骤 1.新建文件夹 2.下载OpenPo ...

  2. OpenSSL在Win10环境下的安装

    Win10环境下OpenSSL的安装 1,前言 2,OpenSSL在Win10环境下的安装 2.1,了解安装OpenSSL所需步骤: 2.2,根据步骤完成前期准备 2.2.1,安装Perl 2.2.2 ...

  3. Python3.5 win10环境下导入kera/tensorflow报错的解决方法

    这篇文章主要介绍了Python3.5 win10环境下导入keras/tensorflow报错的解决方法,较为详细的分析了Python3.5在win10环境下导入keras/tensorflow提示错 ...

  4. cuda9.2环境下pip安装tensorflow非官方轮子

    cuda9.2环境下pip安装tensorflow非官方轮子 对于tensorflow,官方的预编译版本只支持cuda9.0,如果是cuda9.2需要自己编译安装. 但是,github上能够查到热心的 ...

  5. mac下查看redis安装路径_干货!win10环境下Redis安装、启动教程

    " 本文详细介绍win10环境下Redis的安装.启动方法." 01 - 下载 Redis项目不正式支持Windows.但是,Microsoft开放技术小组开发和维护一个Windo ...

  6. win10环境下一键安装TensorFlow-gpu+cuda+cudnn

    这里主要介绍tensorflow-gpu版本的安装,如果需要安装Tensorflow CPU版本的,只需要将以下步骤中的tensorflow-gpu换成tensorflow就行了. 重点!!! 此方法 ...

  7. centos7 mysql安装教程_centos7环境下MySQL安装教程

    Linux centos7环境下安装MySQL步骤的详细介绍 安装MySQL mysql 有两个跟windows不同的地方 1).my.ini 保存到/etc/my.ini 2).用户权限,单独用户执 ...

  8. RabbitMQ 在 Win10 环境下的安装与配置

    文章目录 1. RabbitMQ 环境配置(ErLang 10.4 + MQ 3.7.17) 1.1 ErLang 下载安装 1.2 RabbitMQ 下载安装 1.3 环境变量配置 1.3.1 新建 ...

  9. win10如何安装wxpython_python在win10环境下怎样安装wxpython?

    Python 与 wxPython 运行环境安装 分 别安装python-2.6.2.msi.wxPython2.8-win32-unicode-2.8.10.1-py26.exe wxPython2 ...

最新文章

  1. Redis主从复制(Master-Slave Replication)
  2. sqlserver2008安装报错 “Previous releases of Microsoft Visual Studio 2008″ failed.
  3. C++程序设计:原理与实践(进阶篇)15.6 实例:一个简单的文本编辑器
  4. 使用 bash 脚本把 GCE 的数据备份到 GCS
  5. 决策树算法小结(一) ID3原理及代码实现
  6. shell脚本学习总结02--数组
  7. Git教程 Git Bash详细教程
  8. UE4安装教程,虚幻引擎安装教程,UE4的安装
  9. php全量导出,项目中生成器应用,解决量级数据导出excel内存溢出
  10. 闲谈IPv6-IPv4的TCP和NAT让互联网变得畸形
  11. 013 两个重要极限之一
  12. 目标规划的数学建模及求解
  13. python repr
  14. Android 绘制电池图标
  15. js实现拼图游戏(数字版本与图片版本)
  16. Mac 终于有显示隐藏文件的快捷键了
  17. 支付宝扫码枪流程笔记
  18. 揭开三星手机CPU的神秘面纱
  19. UE4虚幻引擎关于小地图制作教程
  20. ANSVC无功补偿装置的应用--在南京某高等院校中

热门文章

  1. iphone 方向旋转
  2. 自然对数底e的来源1
  3. python 实现ps色阶算法
  4. 计算机条件函数ppt,计算机编程CHP5 函数.ppt
  5. 跨域之nginx本地代理
  6. 之一-呕血制作-Lettuce IOT框架-移远BC35G+树莓派+华为OC+SpringBoot后台+微信小程序
  7. 解决idea中不能输出中文问题
  8. C++/C++11中std::transform的使用
  9. odp 加固 mysql_ODAC (odp.net) 从开发到部署
  10. c语言比例法颜色,颜色空间RGB与 色调、饱和度、亮度(ESL)转换的完整C语言算法...