tensorflow Gpu版本 windows 下的安装

Posted 2021-01-28 lif323

主要参考 tensorflow官方文档 https://tensorflow.google.cn/install/pip?lang=python3

基本环境

显卡: GTX1050ti
操作系统：win10

预先安装的软件

0.软件列表

  打开官网的如下链接：https://tensorflow.google.cn/install/gpu 即可看到需要预先安装的软件。

1.vs2015 专业版 （社区版也可以）安装

  可以到MSDN，我告诉你下载相关镜像。完成相应安装即可。

2. cuda的安装

  到nvidia官网下载cuda9.0版本，这里下载的(local)版本
  版本选择的截图如下：

如果cuda安装过程中出现 “This graphics driver could not find compatible graphics hardware”
这是因为cuda内部的显卡驱动比电脑自身的显卡驱动版本低，可以跳过警告，继续安装。
  

3.cuDnn的安装

  这里的下载需要注册nvidia账号，我安装时的版本选择如下，下载后的到一个.zip压缩文件。
  

官方文档的安装过程

1.python3的安装：（Install the Python development environment on your system）

  tensorflow官网显示只支持python3.4 3.5 3.6 所以需要自行安装满足要求的pyhton3版本，可以下载如下图所示的版本，形化安装比较方便，而且卸载也比较方便，只需要再次运行安装程序即可完成卸载。安装时可以将安装路径添加到系统变量path也可以安装完成后，手动添加。安装路径最好不要在C:Program Files或C:Program Files(X86)路径下，因为安装路径含有空格有可能会造成一些问题。

  官方文档中出现的python3 也就是python 只需要把python安装目录下的python.exe文件改为python3.exe即可。
检测相应的软件是否存在，不存在的话有可能是没有添加到系统变量(添加到系统变量即可)，如果没有安装点击相应的使用pip也很好安装
例如安装virtualenv 执行如下命令:pip install virualenv

3.虚拟环境的创建(Create a virtual environment )：

  按照官方文档说明的步骤就好（俺也一样O_O）

4.tensorflow的安装(Install the TensorFlow pip package)：

  按照官方文档说明的步骤就好（俺还一样o_O）