ApollooxX 2020-04-03 13:18:53
426
收藏 1
最后发布: 2020-04-03 13:18:53 首次发布: 2020-04-03 13:18:53
版权声明:本文为博主原创文章,遵循 CC 4.0 BY-SA 版权协议,转载请附上原文出处链接和本声明。
win10+tensorflow-gpu2.0.0+keras+cuda10.0.130(不可以是 10.1)+7.6.0(要大于 7.4.1)
前提:电脑已经安装了 Anaconda3。
声明:这个版本的 cuda 和 cudnn 已经不需要手动下载安装了,直接命令行通过 conda 即可安装。
额外:如果之前没有更改过 pip 和 conda 镜像源,参考这个:
pip、conda 更改为国内源
正文:
1、首先检查电脑英伟达显卡驱动,键盘 win+r
输入 cmd
打开,输入 nvidia-smi
, 即可看到驱动版本号,需要大于 410.x。如果不够,到英伟达官网下载:https://www.nvidia.cn/Download/index.aspx?lang=cn
2、安装 cuda+cudnn。进入自己的安装环境,我的直接安装在 默认的 base 环境下,打开开始菜单栏的 Anaconda Prompt (Anacanda3),默认是 base 环境下(命令行最前面小括号内就是当前环境名),依次输入:conda install cudatoolkit==10.0.130
和 conda install cudnn==7.6.0
, 安装前会问是否要安装更新,输入 y
即可可以下载安装。
3、再安装 tensorflow-gpu2.0.0。输入 conda install tensorflow-gpu==2.0.0
,即可开始安装。
4、最后安装 keras,,输入 pip install keras
或者 conda install keras
5、查看 tensorflow 以及 cuda 是否安装成功,输入 c:
进入用户目录,输入 python
进入 python 环境,输入 import tensorflow as tf
,显示成功打开一个动态库,再输入 tf.__version__
,显示 '2.0.0
‘表示 TensorFlow2.0.0 安装成功。
输入 tf.test.is_gpu_available()
, 若返回 true
标志可使用 GPU。
https://blog.csdn.net/qq_42902264/article/details/105291024