win10安装tensorflow_gpu

Posted 2020-12-13 xjjsk

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后，只在系统中生效，普通进程是不会生效的，因为它带的是启动时父进程拷贝给他的。所以需要重启进程，环境变量才会更新