Ubuntu安装深度学习环境相关(yolov8-python部署)

  • 本文将从如下几个方面总结相关的工作过程:

    • 1.Ubuntu系统安装(联想小新pro16)
    • 2.显卡驱动安装
    • 3.测试深度学习模型

1. Ubunut 系统安装

  • 之前在台式机上安装过Ubuntu,以为再在笔记本上安装会是小菜一碟,但没想还是废了一些功夫。

  • 安装所需要的步骤:

    • 1.电脑分盘:Windows下右键开始,选择磁盘管理,我的是在F盘分出了80G
    • 2.在- 官网 -下载Ubuntu系统并通过U盘制作启动盘:
    • 在- rufus -下载启动盘制作工具:
    • 点击开始进行制作
    • 我的电脑是联想小新,按下开机建后,同时连续多次按Fn+F2,进入BIOS界面,将Secure Boot关闭,保存更改退出,关机重启
    • 插入U盘启动盘,按下开机键,连续按下F12,选择U盘对应的系统进行安装,选择中文语言,正常下一步,常规安装至结束,中间正常设置账户密码等。
  • 设置中也可以设置输入法:

  • 如果桌面字体太小可以通过改变分辨率进行调整:
    – 打开软件与更新设置国内源

  • 如果系统安装过程最后提示磁盘没有访问权限,需要解密,则需要回到windows下对磁盘进行解密,即通过命令行关闭Bitlocker

  • cmd中输入以下命令关闭

     manage-bde -off C:
  • 但是有时候出现如下提示:

此时需要先执行如下命令:(系统分区不是C的话更改下面的盘符)

manage-bde -autounlock -ClearAllKeys c:

然后再执行即可

manage-bde -off C:

提示解密进行中,需要一定的时间:

解密后所在分区就没有锁子图标了

2.显卡驱动安装

  • 我的显卡是RTX3050,但是我根据网上的一些信息通过软件与更新或者 sudo apt 命令进行下载驱动程序都提示连接不上NVIDIA,后来查了一下说需要先禁用系统自带的驱动程序 Nouveau
  • 查看是否安装NVIDIA驱动命令:
nvidia-smi

2.1 禁用 nouveau

  • a. 运行如下指令,查看系统是否启用了nouveau显卡驱动
lsmod | grep nouveau

如果有内容输出,则说明nouveau已被启用,需要执行后续从b.开始的步骤以将nouveau禁用;如果没有内容输出,则可以直接安装nvidia驱动。

  • b. 修改blacklist.conf配置

blacklist.conf文件的位置:

/etc/modprobe.d/blacklist.conf

不管是通过vi、vim,还是直接修改,选择一种即可,本文采用直接修改的方法进行后续步骤。

Ctrl+Alt+T打开一个新的命令行窗口,通过如下指令切换到blacklist.conf文件所在的目录:

cd ../../etc/modprobe.d 

.conf文件的修改需要管理员权限,需要sudo,通过执行如下指令直接对记事本文件进行修改:

sudo gedit blacklist.conf 

在该文件最后插入以下两行内容:

blacklist nouveauoptions nouveau modeset=0

最终效果如图所示:
保存并关闭该文件。

  • c. 更新系统

通过如下命令更新系统,使刚刚修改的文件生效:

sudo update-initramfs -u
  • d. 重启系统

重新启动系统,可以通过图形界面直接重启,也可以通过命令行重启。本文通过命令行重新启动,在任意命令行中输入如下命令:

reboot

等待系统重启完成。

  • e. 验证nouveau是否已禁用

再次通过如下命令,查看是否有输出:

lsmod | grep nouveau

此时,命令行输出为空,如下图所示。

说明nouveau显卡驱动已被禁用,此时可以安装nvidia显卡驱动。

2.2 安装NVIDIA驱动

  • 官网搜索驱动程序下载:https://www.nvidia.cn/Download/index.aspx?lang=cn
  • 进入下载文件夹,打开命令行窗口:
  • 首先通过如下命令获取该.run文件的执行权限,不然就算下面加了sudo也执行不了。
chmod +x NVIDIA-Linux-x86_64-535.54.03.run
  • 通过如下命令运行下载好的.run文件:
sudo ./NVIDIA-Linux-x86_64-535.54.03.run

注意

“./”后的文件名需要换成在Step 1中根据自己电脑下载的驱动的名称!

安装过程中提示的东西全部安装即可!

遇到的小问题
报错“you appear to be running an x server; please exit x before installing.”

解决方法:在运行命令后加上 --no-x-check

加上之后的命令最终为:

sudo ./NVIDIAxxxx.run --no-x-check
  • 最后执行以下命令看是否安装成功:
nvidia-smi

3.测试深度学习模型

3.1 安装anaconda

  • 清华镜像官网Anaconda下载,我选择的是这个:
    – 下载文件夹打开终端窗口,输入如下命令进行安装,一路yes即可:
bash Anaconda3-2022.05-Linux-x86_64.sh

3.2 搭建虚拟环境

#新建虚拟环境conda create yolov8 python=3.8pip install ultralytics#激活虚拟环境conda activate yolov8#推理:yolo predict model=yolov8n.pt source=./ultralytics/assets/bus.jpg


搞定:

参考:

1. 联想win10安装ubuntu20.04双系统
2.通过命令行关闭Bitlocker
3.Ubuntu安装Nvidia显卡驱动
4.Ubuntu安装Anaconda详细步骤