搭建cuda环境

1 安装驱动

1.1 前期工作

1.1.1 禁用nouveau

ubuntu 16.04默认安装了第三方开源的驱动程序nouveau,安装nvidia显卡驱动首先需要禁用nouveau,不然会碰到冲突的问题,导致无法安装nvidia显卡驱动。指令如下

sudo gedit /etc/modprobe.d/blacklist.conf  打开文件,在最后添加如下两行:

blacklist nouveau

options nouveau modeset=0

1.1.2 更新系统修改

sudo update-initramfs -u      ,输入指令后重启系统(一定要重启),确保到位。

1.1.3 验证nouveau是否已禁用

lsmod | grep nouveau

如果没有出现任何东西就是成功了。

1.2 下载驱动文件并指令安装

1.2.0、重启电脑,关闭secure boot

1.2.1 、在英伟达的官网上查找你自己电脑的显卡型号然后下载相应的驱动: https://www.geforce.cn/drivers,  下载后的run文件拷贝至home目录下,   文件为:NVIDIA-Linux-x86_64-xxx.run

1.2.2 、 在ubuntu下按ctrl+alt+f1进入命令行界面,此时需要login:电脑账户名称,password:密码,登录到命令行界面。 有时会出现登录失败,报错incorrect login ,此时可以按下ctrl+alt+F2(F4)等进入,重新login,即可。

1.2.3、   sudo service lightdm stop      //这个是关闭图形界面,必须关闭

1.2.4、 sudo apt-get remove nvidia-*    //卸载系统中存在的驱动,默认有安装的,一定要执行这个

1.2.5、 sudo chmod  a+x NVIDIA-Linux-x86_64-xxx.run    //给文件权限

1.2.6、   sudo ./NVIDIA-Linux-x86_64-xxx.run -no-x-check -no-nouveau-check -no-opengl-files  ,其中:

-no-x-check:安装驱动时关闭X服务

-no-nouveau-check:安装驱动时禁用nouveau

-no-opengl-files:只安装驱动文件,不安装OpenGL文件

在安装过程中会出现:

1、The distribution-provided pre-install script failed! Are you sure you want to continue?     选择 yes 继续。

2、Would you like to register the kernel module souces with DKMS? This will allow DKMS to automatically build a new module, if you install a different kernel later?            选择NO继续

3. Nvidia's 32-bit compatibility libraries?    选择 No

4、Would you like to run the nvidia-xconfigutility to automatically update your x configuration so that the NVIDIA x driver will be used when you restart x? Any pre-existing x confile will be backed up.  选择 Yes  继续
1.2.7、 安装成功后,在命令行输入:sudo service lightdm start   //重启图形界面,  按Ctrl+Alt+F7返回图形界面

1.2.8、 nvidia-smi  ,   nvidia-settings//检测是否安装成功

到此驱动就安装好了。

2. 安装CUDA

在官网下载相应cuda版本,deb(local)版本,则按照Base installers提示,并配置环境变量

sudo dpkg -i cuda-repo-ubuntu1604-9-2-local_9.2.148-1_amd64.deb
sudo apt-key add /var/cuda-repo-<version>/7fa2af80.pub
sudo apt-get update
sudo apt-get install cuda

以cuda-9.2为例,在~/.bashrc文件最下方添加:

export PATH=/usr/local/cuda-9.2/bin:$PATH
export LD_LIBRARY_PATH=/usr/local/cuda-9.2/lib64$LD_LIBRARY_PATH

激活生效source ~/.bashrc,并运行nvcc -V查看是否安装成功

注意:运行nvidia-smi出现的CUDA版本只需>=安装的cuda版本即可

3. 安装CUDNN

下载tar文件

tar -xzvf cudnn-x.x-linux-x64-v8.x.x.x.tgz
sudo cp cuda/include/cudnn*.h /usr/local/cuda/include 
$ sudo cp -P cuda/lib64/libcudnn* /usr/local/cuda/lib64 
$ sudo chmod a+r /usr/local/cuda/include/cudnn*.h /usr/local/cuda/lib64/libcudnn*

验证:

  1. Copy the cuDNN samples to a writable path.
    $cp -r /usr/src/cudnn_samples_v8/ $HOME
  2. Go to the writable path.
    $ cd  $HOME/cudnn_samples_v8/mnistCUDNN
  3. Compile the mnistCUDNN sample.
    $make clean && make
  4. Run the mnistCUDNN sample.
    $ ./mnistCUDNN
    If cuDNN is properly installed and running on your Linux system, you will see a message similar to the following:
    Test passed!

出现 Test passed即为成功

  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值