美文网首页
在ubuntu16.04 安装 tensorflow-gpu

在ubuntu16.04 安装 tensorflow-gpu

作者: 大数据与人工智能 | 来源:发表于2017-03-22 14:15 被阅读0次

    一. 需要的环境和软件:

    二. 安装步骤:

    1. 查看显卡信息 sudo lshw -numeric -C display
    2. 安装Nvidia 显卡驱动,建议直接在ubuntu 设置中跟新设定为NVIDIA私有驱动
    3. 安装Cuda8.0
    • export PATH=$PATH:/usr/local/cuda-8.0/bin
    • export LD_LIBRARY_PATH=/usr/local/cuda-8.0/lib64
    • nvidia-smi

    4.下载cudnn5.1,并复制到 cuda 安装目录的相应文件夹下
    5 安装tensorflow

    相关文章

      网友评论

          本文标题:在ubuntu16.04 安装 tensorflow-gpu

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