前往小程序,Get更优阅读体验!
立即前往
首页
学习
活动
专区
工具
TVP
发布
社区首页 >专栏 >win10安装tensorflow-gp

win10安装tensorflow-gp

作者头像
py3study
发布2020-01-16 11:55:57
8430
发布2020-01-16 11:55:57
举报
文章被收录于专栏:python3

一,本机配置

Win10 64bit

NVIDIA GeForce GTX 960M

Python3.7(Anaconda)

二,安装CUDA

亲测,TensorFlow-gpu1.13.1支持cuda10.0的版本,所以我们可直接选择cuda10.0的版本

Window10下载CUDA10

安装步骤为:

1)运行.exe可执行和层序。安装软件会先运行一个系统检查,如果没有软硬件不兼容的情况就能继续进行下一步。若有不兼容情况,系统检查则会报错,无法进行下一步。如果报错了,就最先考虑下是不是自己的显卡不被该版本支持。在安装过程中,它会先自动安装和你电脑对应的NVIDIA驱动。 2)选择默认安装就行。

三,安装cuDNN

cuDNN7.3.1下载

下载cuDNN需要自己先在以上网址(cuDNN官网)注册个账号,注册成功后登陆进去选择cuDNN7.3.1下载即可。

下载完成后是一个.zip包,解压后进入目录,会发现三个子目录/bin,/include,/lib/x64。进去后会发现有三个文件,将这三个文件分别拷贝到cuda的安装目录的对应子目录下

默认cuda的安装目录是C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA

四,配置系统Path环境变量

检查你的系统Path环境变量中是否有这四项,没有请手动添加

代码语言:javascript
复制
C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v10.0\bin
C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v10.0\libnvvp
C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v10.0\lib\x64
C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v10.0\extras\CUPTI\libx64

五,安装tensorflow-gpu

打开Anaconda Prompt,输入

代码语言:javascript
复制
pip install tensorflow-gpu

会自动下载并安装tensorflow-gpu

六,安装测试

打开pycharm,创建一个test.py文件,输入

代码语言:javascript
复制
import tensorflow as tf

tf.enable_eager_execution()
print(tf.__version__)
hello = tf.constant("hello tensorflow")
print(hello.numpy())

出现下列结果表明安装成功

本文参与 腾讯云自媒体同步曝光计划,分享自作者个人站点/博客。
原始发表:2019/05/24 ,如有侵权请联系 cloudcommunity@tencent.com 删除

本文分享自 作者个人站点/博客 前往查看

如有侵权,请联系 cloudcommunity@tencent.com 删除。

本文参与 腾讯云自媒体同步曝光计划  ,欢迎热爱写作的你一起参与!

评论
登录后参与评论
0 条评论
热度
最新
推荐阅读
目录
  • 一,本机配置
  • 二,安装CUDA
  • 三,安装cuDNN
  • 四,配置系统Path环境变量
  • 五,安装tensorflow-gpu
  • 六,安装测试
领券
问题归档专栏文章快讯文章归档关键词归档开发者手册归档开发者手册 Section 归档