Anaconda3安装tensorflow-gpu

记录一下自己安装tensorflow的过程,实在遇到了很多坑

我是win10系统

首先安装gpu版tensorflow要安装CUDA和cudNN

tensorflow、CUDA和cudNN和python的版本一定要先查清楚不然就要像我一样重新安装很多遍

自己电脑显卡支持的cuda版本可以按这个步骤:

    桌面右键->NVIDIA控制面板->帮助->系统信息就可以查看适合安装的cuda版本我的显示是10.0.132但是并没有找到那个版本,10.0.130也安不了,但是9.2和9.0版本都可以安装成功

    CUDA下载地址:https://developer.nvidia.com/cuda-toolkit-archive

    cudNN下载地址:https://developer.nvidia.com/rdp/cudnn-archive

    9.2版本的CUDA我选的是7.3.1版本的cudNN

    9.0版本的CUDA我选的是7.1版本的cudNN

    最后用的是 9.0版本的CUDA和7.1版本的cudNN

    下载cudNN的时候就有写明是适合哪个版本的CUDA,照着下早就没问题了

    关于cudNN的安装罗嗦几句

        下载好后解压,里面有三个文件夹:bin、include、lib

        进入文件夹C:->Program Files->NVIDIA GPU Computing Toolkit->CUDA->v9.0

         然后把之前解压的三个文件夹里的文件复制到上面所示路径的对应三个文件夹里面即可

          如果想检验是不是成功了:C:->Program Files->NVIDIA GPU Computing Toolkit->CUDA->v9.0->extras->demo_suite可以运行里面的demo,成功运行即安装成功。

(PS:Nivida我觉得非常迷,9.2版本的CUDA安装gpu版tensorflow怎么试都不行,但是9.0的普适版就很高)

    tensorflow我安装的是1.10.0版本,先安装了Anaconda之后才安装的tensorflow,个人感觉Anaconda是个很强大的工具,功能性很强,用起来也很方便。

    tensorflow的语言是python,安装tensorflow之前也一定要看好版本再进行安装,这是一个对照表

    个人建议选择python3.5,后续安其他的安装包兼容性也比较高 

    好的,接下来安装tensorflow:

        打开anaconda promt,输入:conda create --name tensorflow-gpu python=3.6     创建一个名为tensorflow-gpu的新环境;

   激活:activate tensorflow 

         查看创建的新环境是否被激活: conda info --envs 

         安装指定版本的tensorflow:pip install tensorflow-gpu==1.4.0

         安装好后还需要通过Anaconda Navigator进入tensorflow-gpu环境,安装ipython、spyder、jupyter

          运行程序时还需要安装什么直接pip install就可以了

 

 

 

 

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值