要先查看自己的显卡是否已经安装了驱动
lsmod | grep nouveau   查看源驱动nouveau是否已经载入
sudo ubuntu-drivers devices  查看自己的驱动有哪些选择


开始之前先要查看显卡驱动是否装好。


在系统终端输入命令:glxinfo|grep rendering


如果出现“direct rendering:yes”,则已安装。


软件和更新”,查看“附加驱动”  看看使用的是不是源驱动nouveau,一般是,
1、接下来要卸载源驱动并且安装英伟达的驱动
http://blog.csdn.net/u012581999/article/details/52433609
http://jingyan.baidu.com/article/9f7e7ec04896fb6f281554e7.html
http://blog.csdn.net/misiter/article/details/7652731




1、到该网址https://developer.nvidia.com/cuda-toolkit-archive下载cuda7.5或者cuda8.0




2、安装步骤
如果时.deb文件则
Installation Instructions:


    `sudo dpkg -i cuda-repo-ubuntu1604-8-0-local-ga2_8.0.61-1_amd64.deb`
    `sudo apt-get update`
    `sudo apt-get install cuda`
如果下载的时.run文件则
sudo sh cuda_8.0.27_linux.run
安装过程如果停止在more  可以ctrl+C 
执行后会有一系列提示让你确认,但是注意,有个让你选择是否安装nvidia361驱动时,一定要选择否
其他默认或者是
安装成功后会显示类似如下界面:


===========
= Summary =
===========
Driver:   Not Selected
Toolkit:  Installed in /usr/local/cuda-8.0
Samples:  Installed in /home/jonesky, but missing recommended libraries


Please make sure that
 -   PATH includes /usr/local/cuda-8.0/bin
 -   LD_LIBRARY_PATH includes /usr/local/cuda-8.0/lib64, or, add /usr/local/cuda-8.0/lib64 to /etc/ld.so.conf and run ldconfig as root


To uninstall the CUDA Toolkit, run the uninstall script in /usr/local/cuda-8.0/bin


Please see CUDA_Installation_Guide_Linux.pdf in /usr/local/cuda-8.0/doc/pdf for detailed information on setting up CUDA.


***WARNING: Incomplete installation! This installation did not install the CUDA Driver. A driver of version at least 361.00 is required for CUDA 8.0 functionality to work.
To install the driver using this installer, run the following command, replacing <CudaInstaller> with the name of this run file:
    sudo <CudaInstaller>.run -silent -driver


Logfile is /tmp/cuda_install_4581.log
Signal caught, cleaning up






安装的如果是8.0不需要接下来的环境配置,程序已经自动配置好,直接测试看看
安装后,要声明环境变了,并将其写入到~/.bashrc 的尾部:


export PATH=/usr/local/cuda-8.0/bin${PATH:+:${PATH}}
export LD_LIBRARY_PATH=/usr/local/cuda-8.0/lib64${LD_LIBRARY_PATH:+:${LD_LIBRARY_PATH}}




然后设置环境变量和动态链接库,在命令行输入:
sudo gedit /etc/profile






在打开的文件末尾加入:


export PATH = /usr/local/cuda/bin:$PATH


保存之后,创建链接文件:
sudo gedit /etc/ld.so.conf.d/cuda.conf
在打开的文件中添加如下语句:
/usr/local/cuda/lib64






然后执行
sudo ldconfig
使链接立即生效。




3、测试cuda 的Samples
cd /usr/local/cuda-8.0/samples/1_Utilities/deviceQuery
sudo make
sudo ./deviceQuery
如果显示的是一些关于GPU的信息,则说明安装成功了。




4、下载cudnn,网址为https://developer.nvidia.com/cudnn


a. 首先需要注册一个账号,填一下问卷。登录之后才让你下载,而且下载之前也是简单填个问卷。
b.下载 Download cuDNN v5 (May 27, 2016), for CUDA 8.0 ,注意要和自己安装的cuda的版本对应了。
下载  5.1 library for linux 
已经注册的账号1018841732@qq.com
jonesky889299




下载后解压
tar -zxvf cudnn-8.0-linux-x64-v5.0-ga.tgz 


把相应的文件拷贝到cuda的对应目录。(注意是cuda, 不是cuda-8.0)
sudo cp cuda/include/cudnn.h /usr/local/cuda/include
sudo cp cuda/lib64/libcudnn* /usr/local/cuda/lib64 


这步有的有说要 ,有的没有说
cd /usr/local/cuda/lib64/
sudo rm -rf libcudnn.so libcudnn.so.5    #删除原有动态文件
sudo ln -s libcudnn.so.5.0.5 libcudnn.so.5  #生成软衔接
sudo ln -s libcudnn.so.5 libcudnn.so      #生成软链接




接着d. 修改文件的权限(可有可无)
sudo chmod a+r /usr/local/cuda/include/cudnn.h /usr/local/cuda/lib64/libcudnn*






5、安装 tensorflow-gpu
pip install tensorflow-gpu




6、设置gpu按需使用,theano和tensorflow
参考网址
http://blog.csdn.net/jerr__y/article/details/53695567



Logo

北京人形旗下天工造物具身智能开源社区,聚焦具身天工与慧思开物两大平台

更多推荐