Advertisement

64位Ubuntu14.04+GTX960安装CUDA8.0及cudnn

阅读量:

0.检查gpu,系统版本和内核是否支持,一般没问题。

  • 某些设备的BIOS设置中需要禁用该功能。
  • 遇到黑色屏现象时,请参考这里

1.gcc和g++调整:安装cuda对应的gcc和g++并进行软链接,本教程cuda对应4.8

复制代码
    sudo add-apt-repository ppa:ubuntu-toolchain-r/test
    sudo apt-get update
    sudo apt-get install -y gcc-4.8
    sudo apt-get install -y g++-4.8
    sudo su
    cd ../../usr/bin
    ln -s /usr/bin/g++-4.8 /usr/bin/g++ -f
    ln -s /usr/bin/gcc-4.8 /usr/bin/gcc -f

2. 选择安装32位兼容版本软件。在二进制软件安装过程中,默认情况下不会提供32bit兼容选项,默认选择通常会导致后期操作上的诸多不便。因此建议主动选择并完成该步骤,在后续的操作系统更新中更容易配置和管理这些软件。

复制代码
    sudo apt-get install -y lib32z1

3.禁用nouveau

复制代码
    lsmod | grep nouveau   查看nouveau是否使用,无输出则意味着被禁用或没有
    有则执行sudo vim /etc/modprobe.d/blacklist-nouveau.conf, 输入
    blacklist nouveau
    options nouveau modeset=0
    保存退出
    在终端执行sudo update-initramfs -u
    重启电脑

4.安装CUDA

复制代码
    sudo service lightdm stop 关闭图形界面,按ctrl+alt+F1进入tty1并登录
    进入run文件所在目录
    chmod 777 cuda_8.0.61_375.26_linux.run
    sudo ./cuda_8.0.61_375.26_linux.run 需要时间读取许可,按q退出阅读许可
    依次接受许可,安装driver:yes,安装toolkit和samples:yes,中间选择及安装路径直接Enter选择默认
    sudo service lightdm start 打开图形界面
    执行vim  ~/.bashrc,在文件最后加入
    export PATH=/usr/local/cuda/bin:$PATH  
    export LD_LIBRARY_PATH=/usr/local/cuda/lib64:$LD_LIBRARY_PATH
    执行source ~/.bashrc使配置的环境变量生效 
    nvidia-smi 出现GPU相关信息表示驱动安装成功
    执行nvcc --version查看cuda版本信息

5.测试cuda的Samples

复制代码
    cd /usr/local/cuda-8.0/samples/1_Utilities/deviceQuery  
    sudo make  
    sudo ./deviceQuery
    保证编译没有error,执行deviceQuery输出信息的最后类似以下:
    deviceQuery, CUDA Driver = CUDART, CUDA Driver Version = 8.0, CUDA Runtime Version = 8.0, NumDevs = 1, Device0 = GeForce GTX 960
    Result = PASS

6.cudnn安装

复制代码
    tar xvzf cudnn-8.0-linux-x64-v5.1.tgz  
    sudo cp cuda/include/cudnn.h /usr/local/cuda/include  
    sudo cp cuda/lib64/*.* /usr/local/cuda/lib64 -a
    sudo chmod a+r /usr/local/cuda/include/cudnn.h /usr/local/cuda/lib64/libcudnn*

7.卸载

复制代码
    sudo nvidia-uninstall 或者
    sudo ./cuda_8.0.61_375.26_linux.run --uninstall 或者
    sudo apt-get remove --purge nvidia-* 卸载驱动
    sudo /usr/local/cuda-8.0/bin/uninstall_cuda_8.0.pl 卸载cuda

全部评论 (0)

还没有任何评论哟~