美文网首页
Ubuntu 16.04安装N卡驱动、cuda和tensorfl

Ubuntu 16.04安装N卡驱动、cuda和tensorfl

作者: 佩鸿PH | 来源:发表于2017-11-17 22:39 被阅读0次

    安装驱动

    最开始在英伟达官网下载了官方驱动,安装之后无法登录系统,在登录界面反复循环,用cuda里的驱动也出现了同样的问题。最后解决办法是把驱动卸载之后,通过命令行在线安装驱动。
    卸载驱动:

    sudo nvidia-uninstall
    

    在线安装:

    sudo apt-add-repository  ppa:graphics-drivers/ppa
    sudo apt-get update
    sudo apt-get install nvidia-384 #写这篇文章时,最新的N卡驱动是384,具体驱动版本号可以到官网查
    

    安装完之后,在终端输入

    nvidia-smi
    

    出现下图,说明显卡安装成功:


    安装cuda

    在英伟达官网下载cuda安装包(.run文件),在终端中切换到cuda安装包目录,直接sudo运行安装包即可。注意在问你是否要安装显卡驱动的时候,选择no。
    安装之后,添加环境变量:

    gedit ~/.bashrc
    

    在文件最后添加如下几行:

    export PATH="/usr/local/cuda/bin:$PATH"
    export CUDA_HOME="/usr/local/cuda"
    export LD_LIBRARY_PATH="/usr/local/cuda/lib64:/usr/local/cuda/extras/CUPTI/lib64:$LD_LIBRARY_PATH"
    

    安装tensorflow

    安装之前需要安装libcupti-dev库,

    sudo apt-get install libcupti-dev
    

    安装tensorflow的方法很多,最简单的方法是用pip(或者用conda命令,如果你安装了anaconda的话)安装:

    pip install tensorflow      # Python 2.7; CPU support (no GPU support)
    pip3 install tensorflow     # Python 3.n; CPU support (no GPU support)
    pip install tensorflow-gpu  # Python 2.7;  GPU support
    pip3 install tensorflow-gpu # Python 3.n; GPU support 
    

    亲测,下载速度太慢.下面介绍anaconda的安装方法。需要先安装上anaconda,里面包含了python以及大部分常用的科学计算工具包,例如numpy、pandas、matplotlib等。然后通过如下指令安装:

    conda create -n tensorflow python=2.7 # or python=3.3, etc.
    source activate tensorflow
    pip install --ignore-installed --upgrade tfBinaryURL
    

    tfBinaryURL是tensorflow二进制文件链接,不同版本的链接不同,例如安装python2.7 GPU版本:

    pip install --ignore-installed --upgrade https://storage.googleapis.com/tensorflow/linux/gpu/tensorflow_gpu-1.4.0-cp27-none-linux_x86_64.whl
    

    其他版本链接地址请参考:https://www.tensorflow.org/install/install_linux#the_url_of_the_tensorflow_python_package(要翻墙)
    验证安装是否成功,在python命令行中输入如下代码:

    import tensorflow as tf
    hello = tf.constant('Hello, TensorFlow!')
    sess = tf.Session()
    print(sess.run(hello))
    

    如果输出Hello, TensorFlow!,说明安装成功。如果报错,请参考如下链接:
    https://www.tensorflow.org/install/install_linux#common_installation_problems

    参考资料:

    [1] https://www.tensorflow.org/install/install_linux

    相关文章

      网友评论

          本文标题:Ubuntu 16.04安装N卡驱动、cuda和tensorfl

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