ubuntu16.04深度学习环境搭建一:Ubuntu16.04+cuda10.0+cudnn+GTX1660ti环境配置
Ubuntu16.04+cuda10.0+cudnn+GTX1660ti环境配置
(ubuntu16.04深度学习环境搭建一)
Ubuntu下搭建深度学习环境真是一波三折,一开始按照网上的一些博客安装,花了两天的时间把系统安崩了,一直卡在登录界面循环!痛下心将好不容易安好的驱动卸载,参考一篇看着朴实实则十分有效的博客将这部分环境全部搭建好(https://blog.****.net/SWUSTZ/article/details/100559168),且非常的快!按照经历的操作步骤,依次为:
一,安装GTX1660ti驱动:
1. 下载显卡驱动。到Nvidia官网(https://www.geforce.cn/drivers)进行下载。在下图所示的页面中可以根据手动搜索自己所需的显卡驱动,如果可以科学上网,最好采用科学上网访问,此时下载相应的驱动也比一般情况下快很多!
2. 检测内核文件:
sudo apt-get install linux-headers-$(uname -r)
3. 禁止nouveau驱动:
sudo vim /etc/modprobe.d/blacklist.conf
(或者sudo gedit /etc/modprobe.d/blacklist.conf)
4. 在文档末尾写入:
blacklist nouveau
options nouveau modeset=0
5. 更新内核:
sudo update-initramfs -u
6. 重启:
reboot
7. 检测是否禁用成功:
lsmod | grep nouvea
8. 通过ctrl+alt+f2进入终端,登入账户,通过下面命令关闭图像界面:
sudo service lightdm stop
9. 切换到之前下载好的驱动文件目录下,我将下载好的驱动文件放在了home的文件夹nvidia-driver下:
cd /home/你的用户名/nvidia-driver/
10. 执行:(版本号仅供参考)
sudo bash NVIDIA-Linux-x86_64-430.40.run -no-opengl-files
11. 安装过程中会有一些选项,下图这个问题有些博主选No,有些选Yes,根据上文提到的参考博客,我也选了No:
其他问题可参考博客(https://blog.****.net/wangyjfrecky/article/details/84029668)中的选项。
12. 安装完成后,开启图像界面:
sudo service lightdm start
13. 查看驱动是否安装成功:
nvidia-smi
如果出现驱动信息列表则说明安装成功,如下图所示:
二,安装cuda10.0:
1. cuda10.0同样采用科学上网的方式到官网下载,根据系统的配置进行选择,此处选择runfie(local)的方式进行安装:
2. 到下载文件对应的目录下运行终端。执行:
sudo sh cuda_10.0.130_410.48_linux.run
3. 在安装界面中一直按住空格键能够快速跳到底部,选择accept。
需要注意的是,有一个选择为是否安装默认驱动,此处选择否,因为我们前面已经安装好了驱动,所以这里不需要安装,其他的选择默认选项即可。
4. 安装完成后,会在home目录下生成一个NVIDIA_CUDA-10.0_Samples的样例,cd进入这个目录(或者进入该目录后点击在终端打开),执行编译:
make -j8
5. 编译完成后,执行:
cd bin/x86_64/linux/release/
./deviceQuery
如果最后出现result=pass表示样例通过
6. 配置环境变量:
打开~/.bashrc文件:sudo vim ~/.bashrc (或者sudo gedit ~/.bashrc)
将路径写入到文件尾部:
export LD_LIBRARY_PATH=/usr/local/cuda/lib64:$LD_LIBRARY_PATH
export PATH=/usr/local/cuda/bin:$PATH
7. 载入配置:
source ~/.bashrc
8. 查看版本,如果出现版本信息则表示配置成功:
nvcc -V
三,安装cudnn:
1. 到官网(https://developer.nvidia.com/rdp/cudnn-download)下载与cuda10.0对应的cudnn版本,这里是cudnn7.6.5,选择cudnn library for linux进行下载,此处下载的是.solitairtheme8文件,到文件所在的目录下运行终端,先执行以下命令将.solitairtheme8转换为.tgz,再进行解压:
cp cudnn-10.0-linux-x64-v7.6.5.32. solitairetheme8 cudnn-10.0-linux-x64-v7.6.5.32.tgz
tar -xvf cudnn-10.0-linux-x64-v7.6.5.32.tgz
2. 把cudnn的头文件和库拷贝到cuda下的include和lib64下:
sudo cp cuda/include/cudnn.h /usr/local/cuda/include/
sudo cp cuda/lib64/libcudnn* /usr/local/cuda/lib64/
sudo chmod a+r /usr/local/cuda/include/cudnn.h
sudo chmod a+r /usr/local/cuda/lib64/libcudnn*
3. 通过运行以下命令来查看cudnn是否安装成功:
cat /usr/local/cuda/include/cudnn.h | grep CUDNN_MAJOR -A 2
如果出现版本信息,则表示安装成功:
补充:
这里补充一点,是我在安装cuda时重启电脑后一直卡在登录界面,循环登录。当时不想卸载驱动,毕竟是费时费力才安装好的驱动,尝试了多种方法后,一狠心将驱动卸了,循环登录问题被解决。后来在网上找到上文提到的看着朴实但操作起来十分给力的博客(https://blog.****.net/SWUSTZ/article/details/100559168),不到一天的时间就把环境搭建好了(之前被卸掉的驱动我安装了一整天,网速慢也是原因之一,重要的还是安装方法)。参考博客(https://blog.****.net/weixin_43981221/article/details/90113079)卸载驱动:
1. 按ctrl+alt+F2切换到终端,输入自己的登录名和密码
2. 卸载nvidia驱动:sudo apt-get remove --purge nvidia*
3. 重启:sudo reboot
参考博客:
https://blog.****.net/lihe4151021/article/details/90237681
https://blog.****.net/SWUSTZ/article/details/100559168
https://blog.****.net/wangyjfrecky/article/details/84029668
https://blog.****.net/weixin_43981221/article/details/90113079