Ubuntu装机步骤与环境配置(Nidia+CUDA+cudnn+Tensorflow+Opencv+Pytorch)

狂风中的少年 提交于 2019-12-02 12:45:11

1、从镜像安装ubuntu18.04

2、安装完成后的一些设置

  • 更换源:设置-软件和更新-下载自-其他站点-mirrors.ustc.edu.cn

  • sudo apt remove ristretto pidgin pidgin-otr fcitx-data fcitx-sunpinyin thunderbird orage libreoffice-writer libreoffice-calc libreoffice-math libreoffice-core libreoffice-common mousepad

  • sudo apt update

  • sudo apt install axel gedit fcitx fcitx-googlepinyin g++ hardinfo cmake apt-file psensor unrar unzip p7zip-full vim filezilla guvcview i7z eog ttf-wqy-zenhei shutter git

  • apt-mark hold xxxx #忽略安装某些软件,例如gitlab-ce、flashinstaller等下载缓慢或版本不兼容的软件,恢复时用apt-mark unhold xxx

  • sudo apt upgrade && sudo apt dist-upgrade && sudo apt autoremove

3、常用配置

#开关机显示详细信息(错误排查)
sudo gedit /etc/default/grub
#删除quiet splash这句
sudo update-grub
 
#Grub引导菜单记住上一次选择的系统,添加下面两句
gedit /etc/default/grub
GRUB_DEFAULT=saved
GRUB_SAVEDEFAULT=true
sudo update-grub
 
#编译C/C++时彩色显示警告、出错信息,添加下面两句
gedit ~/.bashrc
alias gcc='gcc -fdiagnostics-color=auto'
alias g++='g++ -fdiagnostics-color=auto'
source ~/.bashrc
#如需要sudo时生效,将上面两句话加到sudo vim /etc/profile中,重启后生效
 
#终端中输入字符之后,就可以用方向键Up,Down来搜索以该串字符开头的历史命令
gedit ~/.bashrc
bind '"\e[A": history-search-backward' 
bind '"\e[B": history-search-forward'
source ~/.bashrc
 
#在Ubuntu下关闭UTC(双系统与windows时间同步)
sudo apt install ntpdate
sudo ntpdate time.windows.com
sudo hwclock --localtime --systohc
 
#开启ntfs硬盘写权限
需要在Windows10系统中关闭快速启动功能,打开cmd,运行下面指令
powercfg /h off

4、安装nvidia驱动

       首先终端输入lsmod | grep nouveau,如果显示nouveau的信息,则要先将nouveau禁用,方法如下:

sudo gedit /etc/modprobe.d/blacklist.conf
在末尾添加下面两句话:
blacklist nouveau
options nouveau modeset=0
保存并关闭,然后执行 sudo update-initramfs -u

       重启后用 lsmod | grep nouveau 查看,如果没有任何输出则说明禁用成功。

       然后,点击设置--软件和更新--附加驱动--选择对应的nvidia驱动--应用更改--输入密码--等!

       完成后重启电脑,在终端输入指令 nvidia-smi 查看,显示nvidia的信息则表示安装驱动成功。

5、安装CUDA

        进入BIOS--Boot--Secure Boot--Disabled,重启。

        在官网下载好对应版本的cuda的.run文件,cd到下载目录,执行

   sudo sh cuda_XXXXXX_linux.run

        按提示进行安装,注意不要选择nvidia驱动安装项(x表示选择,空白表示不选择)

        sudo gedit ~./bashrc

        将下面两句添加到末尾并保存
       

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

        然后,source ~/.bashrc

        可通过  nvcc -V 或者 cat /usr/local/cuda/version.txt  查看CUDA版本。

6、安装cudnn

下载对应cuda版本的cudnn,需要注册Nvidia开发者账号。https://link.jianshu.com/?t=https://developer.nvidia.com/cudnn

下载cuDNN5.1之后切换到下载目录进行解压:$sudo tar -zxvf ./cudnn-8.0-linux-x64-v5.1.tgz

进入cuDNN5.1解压之后的include目录,在命令行进行如下操作:
 
cd cuda/include
sudo cp cudnn.h /usr/local/cuda/include #复制头文件
 
再将进入lib64目录下的动态文件进行复制和链接:(libcudnn的版本根据自己情况调整)
 
cd ..
cd lib64
sudo cp lib* /usr/local/cuda/lib64/ #复制动态链接库
cd /usr/local/cuda/lib64/
sudo rm -rf libcudnn.so libcudnn.so.5 #删除原有动态文件
sudo ln -s libcudnn.so.5.1.10 libcudnn.so.5 #生成软衔接
sudo ln -s libcudnn.so.5 libcudnn.so #生成软链接
sudo ldconfig #使配置生效
 
测试CUDA的samples
 
cd /usr/local/cuda-10.0/samples/1_Utilities/deviceQuery
sudo make
sudo ./deviceQuery
 
如果显示一些关于GPU的信息,则说明安装成功。

通过cat /usr/local/cuda/include/cudnn.h | grep CUDNN_MAJOR -A 2 查看cuDNN版本

 

 

 

标签
易学教程内所有资源均来自网络或用户发布的内容,如有违反法律规定的内容欢迎反馈
该文章没有解决你所遇到的问题?点击提问,说说你的问题,让更多的人一起探讨吧!