关于安装tensorflow-gpu

版权声明:本文为博主原创文章,未经博主允许不得转载。 https://blog.csdn.net/california94/article/details/83590883

       安装tensorflow-gpu重点一定型号匹配。我的显卡是NVIDIA Quadro P4000。软件安装的是CUAD9.0  CUDNN9.0  python3.6.5

tensorflow-gpu 1.8.0 。

        1、首先安装python,网上一大堆就不细说了。显卡驱动记得安装!!!!

        2、CUDA下载地址:https://developer.nvidia.com/cuda-toolkit-archive  ,下载的时候选对版本和型号,我下载的cuda9.0,然后安装到电脑上。在电脑环境path里加入你cuda安装路径下的bin路径 和lin\x64路径。

        3、CUDNN下载地址: https://developer.nvidia.com/rdp/cudnn-download ,下载相对应的版本,我下载的cudnn9.0。可能下载的时候需要注册一下。下载后是压缩包,解压后

将这三个文件夹里的文件复制到 C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v9.0 (如果你安装cuda9.0的时候路径没有改过的话是这个路径,如果改动了,找到NVIDIA GPU Computing Toolkit文件夹路径,再往下找cuda v9.0)

将文件复制到相应的位置。如bin的文件复制到这个bin下。

       4、复制完成后安装tensonflow-gpu。 在cmd里运行 pip install tensorflow-gpu==1.8.0 安装完后测试一下。

进入到python命令行中, import tensorflow即可。若是不报错说明没问题了。若是不放心,网上找个测试例子。

    ps:若是报错说缺少 某dll文件,根据提示的错误信息解决。我当时因为缺少的是编译c++的环境,去微软官网下载Microsoft Visual C++组件,根据系统版本选择。

猜你喜欢

转载自blog.csdn.net/california94/article/details/83590883