Win10+Python3.7.6+tensorflow_gpu-2.2.0+CUDA10.1+cuDNN7.6.4.38

0\说明

这是跟着大佬“年轻的li-sir哟”的文章“win10+python3.7.6安装tensorflow踩坑详解”做了一遍后的一些自己的经验总结,大佬讲的更详细链接如下:
https://blog.csdn.net/qq_41912098/article/details/106032630

1\根据python版本及是否需要GPU选择Tensorflow版本

我采用的套餐是——python3.7.6 + tensorflow_gpu-2.2.0

2\根据Tensorflow版本选择CUDA以及cuDNN

可在如下这个网址查询(需要点魔法):
https://www.tensorflow.org/install/source_windows

可我是麻瓜。。。
不过大佬“年轻的li-sir哟”的文章中贴心地给出了截图,文章链接如下(windows版):
win10+python3.7.6安装tensorflow踩坑详解
另一位大佬“佛前一朵莲”在2020年6月23日给出了更新的截图,文章链接如下(注意操作系统):
tensorflow各个版本与cuda版本的对应关系

我采用的套餐是——tensorflow_gpu-2.2.0 + cuDNN7.6.4.38 + CUDA10.1
(虽然版本与表格中不同,但cuDNN7.6.4.38和CUDA10.1在我家没闹别扭)

3\下载CUDA10.1后安装

英伟达官网下载链接:
https://developer.nvidia.com/cuda-toolkit-archive

在这里插入图片描述
在这里插入图片描述
下载后的文件是:
cuda_10.1.243_426.00_win10.exe
按照大佬“年轻的li-sir哟”的文章中说的,关闭杀毒软件,选择精简安装,然后一直下一步到完成

4\下载与安装cuDNN7.6.4.38

即使NVIDIA注册了,下载下来的东西也不完整无法解压,用记事本打开下载后的压缩文件内容为:
在这里插入图片描述
但是!另一位大佬“zhangxiao123qqq”的文章里有盘上资源,文章链接如下:
CUDA10 下载百度云 cuDnn下载(Windows版 )

下载下来的文件是:
cudnn-10.1-windows10-x64-v7.6.4.38.zip

然后就可以顺利地按大佬“年轻的li-sir哟”的文章中说的,找到CUDA文件夹
(我的是C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v10.1)

把刚下载的压缩包中的内容特殊的对应关系复制到安装好的CUDA的相应文件夹中:
压缩包的bin中的内容——复制到——>CUDA的bin文件夹中
压缩包的include中的内容——复制到——>CUDA的include文件夹中
压缩包的lib中的内容——复制到——>CUDA的lib文件夹中

5\用镜像源安装tensorflow_gpu-2.2.0

pip install -i https://pypi.mirrors.ustc.edu.cn/simple/ tensorflow-gpu==2.2.0

6\测试GPU版的tensorflow是否成功安装

在进入python后输入

import tensorflow as tf

print(tf.test.is_gpu_available())

如没有报错,而是出现一大堆代码后如下图片,则OK啦!
在这里插入图片描述

  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值