码迷,mamicode.com
首页 > Windows程序 > 详细

win10安装tensorflow_gpu

时间:2020-06-30 11:18:46      阅读:79      评论:0      收藏:0      [点我收藏+]

标签:nump   必须   byte   进程拷贝   访问   googl   下载   英伟达   python   

1:打开以下网址,找到你要安装的tensorflow版本,以及这个版本对应的其他依赖的版本。依赖版本必须对应上,否则tensorflow会不正常。

https://tensorflow.google.cn/install/source_windows?hl=zh_cn

https://tensorflow.google.cn/install/source?hl=zh_cn#linux

我的环境是win10, 1070显卡,安装tensorflow_gpu-2.0.0,于是找到了下面这条。我不是源码编译安装,所以不需要bazel工具。

技术图片

 

按照指示,去英伟达开发者网站下载CUDA和cuDNN,需要访问外网,否则下载下来是不可用的一个40多Byte小文件。下载cuDNN需要注册账号,填写一些简单的信息。

CUDA下载后直接双击安装。

cuDNN下载后是一个压缩包,解压后放在一个固定目录,然后将解压后的目录下的bin目录添加到环境变量PATH中。

2:python的安装,相信你肯定已经有python了,我也是。这里需要将自己的python卸载掉,然后下载ANACONDA,使用ANACONDA自带的python。如果不卸载你的python,安装ANACONDA后会存在2个python,最终导致包错乱,这是我遇到的一个问题。

3:确保你的系统只有一个python后,安装tensorflow:

pip install tensorflow_gpu;非常慢,最后read timeout。所以需要换一个pip源,网上有很多教程,随便换一个就行。换了之后,pip install tensorflow_gpu一分钟完成。

4:安装完成,但是当你import使用时,还可能会报错,我遇到的错误是winError 193,这个错误是32位和64位不匹配导致,仔细看报错堆栈信息里面带的那些库,发现一个numpy字样,怀疑是numpy是32位的,于是把pip uninstall numpy;pip install numpy;这样卸载再安装,numpy变成64位的了,import成功。

 

细节提示:系统PATH环境变量,在程序启动时,会从父进程传给子进程。所以当你修改了PATH后,只在系统中生效,普通进程是不会生效的,因为它带的是启动时父进程拷贝给他的。所以需要重启进程,环境变量才会更新

win10安装tensorflow_gpu

标签:nump   必须   byte   进程拷贝   访问   googl   下载   英伟达   python   

原文地址:https://www.cnblogs.com/xjjsk/p/13212221.html

(0)
(0)
   
举报
评论 一句话评论(0
登录后才能评论!
© 2014 mamicode.com 版权所有  联系我们:gaon5@hotmail.com
迷上了代码!