美文网首页我爱编程
Tensorflow-GPU版本-路过的坑

Tensorflow-GPU版本-路过的坑

作者: 我去领个便当 | 来源:发表于2017-09-24 18:23 被阅读0次

    用了一天的时间终于把GPU版本的Tensorflow安装成功,其中几个需要注意的事情;

    1、电脑为Win7 64系统,显卡Nvida GT740M;

    2、根据显卡到:https://developer.nvidia.com/cuda-downloads下在cuda,下载时候注意电脑的系统以及版本;

    3、下载cudnn-8.0-windows7-x64-v6.0,注意cudnn要和cuda的版本对应,这是其中一个坑;

    4,、安装好cuda之后,解压cudnn,将其中的三个文件放到cuda的对应文件夹内(cuda的安装目录),自己的是:C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA

    5、使用anaconda,目前使用的anaconda4.3版,支持python3.6,但是目前tensorflow-gpu版本只支持python3.5,通过conda命令建立python3.5的虚拟环境:conda create -n tensorflow python=3.5 anaconda,anaconda有个好处,建立此环境后,在anacond软件快捷键中会有下图Anaconda Prompt(tensorflow)的提示,使用时候,直接点击就可以进入该环境下;

    总结一下:网上的关于win7系统安装tensorflow-gpu的教程很多,就不重复写了;

    之前一直安装失败,出现下图的结果:

    根据网上教程,主要需要注意:

    1、安装cuda要和电脑系统对应,安装cudnn要和cuda的版本对应;安装cuda和cudnn之后一定要测试cuda是否安装成;如果测试过程报错,很可能是显卡驱动不是最新的,更新显卡驱动;(前提条件:cuda以及cudnn的版本对应都是正常的)

    2、通过anaconda建立python3.5的虚拟环境。

    相关文章

      网友评论

        本文标题:Tensorflow-GPU版本-路过的坑

        本文链接:https://www.haomeiwen.com/subject/motlextx.html