win10下 tensorflow gpu版本安装

刚买了带有GPU版本的电脑,其实想直接在ubuntu下安装,因为有服务器,就不想麻烦了,在windows下装看看。
感觉版本不兼容挺烦的,网上博客不是很靠谱,每个人版本都不一样,最靠谱的还是官网说明。
在这里插入图片描述
在这里插入图片描述
在这里插入图片描述
这三张图我也不是看的官网,来自博客地址:https://blog.csdn.net/yeler082/article/details/80943040。
下次直接看官网就好了。。

看要求,以为需要安装VS2015,其实不用。

windows10 64bit,gpu是机器学习最入门的配置GeForce GTX 1050 Ti。我选择的是上图windows版本的tensorflow_gpu-1.3.0,python 3.6,cudnn 6,cuda 8。选择安装这个tensorflow版本,就不能安装其他版本的cuda和cudnn了,确实一开始看博客,安装了cuda 9,或者cudnn 5.1,都报错了,一选择这个配置安装后,就一点问题没有。。

步骤:

  1. 安装 Anaconda,这里选的是(Anaconda3-5.0.1-Windows-x86_64)对应的是python3.6,暂时不能下最新的Anaconda,因为它是python 3.7,而tensorflow只兼容到python 3.6。下好了一步步安装就好了,不需要安装在C盘,记得添加到PATH里,就是安装时有让两个前面需要打钩的选项,都要打上勾。下载Anaconda地址:https://repo.anaconda.com/archive/
  2. 安装 Cuda,这里选的是cuda 8.0 GA2,下好了一步步安装就好了。下载cuda地址:https://developer.nvidia.com/cuda-toolkit-archive。
    在这里插入图片描述
  3. 安装 cudnn,选的是cuDNN v6.0 (April 27, 2017), for CUDA 8.0,下好了,解压后有三个目录,分别添加到cuda路径下的三个目录中去,比如我的cuda安装目录为C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v8.0,要把cudnn的bin、include、lib中的三个文件分别添加到cuda下的对应目录中去。cudnn 下载地址:https://developer.nvidia.com/rdp/cudnn-archive。
  4. 安装 tensorflow,gpu版本,这里选的是tensorflow_gpu-1.3.0,tensorflow,这里选的是tensorflow_gpu-1.3.0,tensorflow,安装方法:cmd到命令行,输入 pip install tensorflow_gpu==1.3.0

安装完毕,可以在cmd中输入 nvcc -V查看cuda是否安装上,然后在spyder或pycharm中 输入 import tensorflow 看是否报错,没报错说明安装成功。

猜你喜欢

转载自blog.csdn.net/eqiang8848/article/details/84921683