欢迎您访问程序员文章站本站旨在为大家提供分享程序员计算机编程知识!
您现在的位置是: 首页

tensorflow-gpu的安装

程序员文章站 2022-07-04 20:08:13
...

配置

首先讲一下我的电脑配置
windows7 64位
显卡 GTX 1660 6G
python3.7
CUDA10.0
CUDNN7.6
tensorflow-gpu1.13.1
pycharm2018(这个版本可以**到2100年,后面的版本貌似不行)

需求

整个过程需要下载CUDA tookit 10.0,CUDNN 7.6以及最后pip install tensorflow-gpu1.13.1

过程

因为之前用的是tensorflow-cpu,所以要先卸载掉cpu版本的tensorflow

pip uninstall tensorflow

之后要查看显卡支持的CUDA版本
1—打开NVIDIA控制面板
tensorflow-gpu的安装
点帮助→系统信息→组件
tensorflow-gpu的安装
这个10.2.95就是支持的CUDA最高版本,确定了支持的CUDA版本,那么就可以去官网下载CUDA了
链接:CUDA Tookit

在这里点击寻找旧版
tensorflow-gpu的安装
在这一步踩了第一个雷,我直接下载了CUDA10.2,因为tensorflow目前不支持CUDA10.2版本,所以最后卸载掉又装了CUDA10.0
tensorflow-gpu的安装
下载时不要直接点这个Dowload 右键点击它→复制链接地址,之后粘贴到迅雷里面下载会快一点。

下载以后打开安装包直接下一步下一步下一步就可以,安装位置就选默认的位置,放在C盘。

之后检验CUDA是否安装成功:打开cmd输入:

nvcc -V

回车就可以查看CUDA版本tensorflow-gpu的安装
这个样子就是安装成功了

之后安装CUDNN,找CUDNN的时候直接去官网找就行,CUDNN要对应上CUDA的版本,不需要对应上tensorflow的版本(根据CUDA版本选择CUDNN和tensorflow)
链接:CUDNN Archive

打开以后找到适合自己CUDA版本的CUDNN ,直接点下载的话会要求注册登录,比较麻烦,老规矩,右键复制下载链接到迅雷里面下载就好。

下载好了解压之后是这样
tensorflow-gpu的安装
然后把bin、include、lib这三个文件夹复制粘贴到下面路径覆盖

C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v10.0

CUDNN安装成功

之后就是去安装tensorflow-gpu了,查找了对照表之后选择了tensorflow1.13.1版本
链接:tensorflow版本参考
tensorflow-gpu的安装
之后就是安装tensorflow-gpu了

pip install -i https://pypi.tuna.tsinghua.edu.cn/simple tensorflow-gpu==1.13.1

至此,安装成功。
检验是否成功:

import tensorflow as tf
hello = tf.constant('hello,tensorflow!')
sess = tf.Session()
print(sess.run(hello))

没有报错即为成功

以上是我自己安装tensorflow的过程及理解,如有问题请指出。

相关标签: tensorflow cuda