美文网首页
win10 tensorflow gpu anaconda 安装

win10 tensorflow gpu anaconda 安装

作者: 在做算法的巨巨 | 来源:发表于2018-11-17 14:44 被阅读0次

    坑是踩不完的,我也是一个雷一个雷踩过来。。。晕

    电脑配置

    • Anaconda 4.3.30
    • python 3.6.5
    • windows 10 profession
    • 显卡NVIDA Quadro K620

    基础操作参考,https://blog.csdn.net/qq_30611601/article/details/79067982
    还是比较清楚的。
    因为我自己已经安装好了Anaconda,所以现在就只需要安装tensorflow。

    • 第一坑,概念坑,tensorflow for CPU or for GPU。
      如果CPU,很简单,pip install tensorflow
      如果GPU, 就回归本文。

    • 第二坑,版本坑
      百度你的显卡支持不支持,也可以参考这个链接https://developer.nvidia.com/cuda-gpus
      如果支持,下载CUDA toolkit,不建议下载最新版,我下载了出现兼容问题,所以重新下载8.0版本,官网下载地址


      两份文件都下载,先安装第一份,安装完成后,再安装第二个。
    • 第三坑 环境坑
      我们装tensorflow_gpu需要一个单独的环境,这个概念就和版本一样,有的python装了2.6还装了3.6,用的时候切换语言版本,一个概念。也就是说,tensorflow_gpu我们需要把我们用到的anaconda重新在新环境下安装一遍。

    1. 进入cmd管理员权限界面。
    2. 创建环境
     conda create -n tensorflow_gpu python=3.6.5
    

    这里提示我们如果启用这个环境需要运行activate,不用deactivate。

    1. 启用tensorflow_gpu
    activate tensorflow_gpu
    

    然后对话框变化出现我们新的环境---tensorflow_gpu。

    1. 接下来安装新环境的anaconda。
    conda install anaconda
    
    1. 安装tensorflow
    conda install tensorflow
    conda install tensorflow-gpu
    
    1. 确认tensorflow是否安装成功,命令框输入以下代码
    import tensorflow as tf
    hello = tf.constant('Hello,TensorFlow!')
    sess = tf.Session()
    print(sess.run(hello))
    

    如果成功输出,就结束了吗,naive
    接下来,你会发现在anaconda里的spyder也好或者jupyter notebook还是导入tensorflow。
    为什么?
    环境没有选择,我们的tensorflow安装在tensorflow_gpu的环境下,因此你需要启动其他app之前先选择环境。



    点击会出现我们所有的环境。
    选择tensorflow_gpu

    最终jupyter notebook上的实现


    相关文章

      网友评论

          本文标题:win10 tensorflow gpu anaconda 安装

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