当前位置:首页 > 系统教程 > 正文

踩坑指南:笔记本Ubuntu 20.04环境搭建

踩坑指南:笔记本Ubuntu 20.04环境搭建

Nvidia RTX 5060驱动+CUDA+PyTorch+ROS实现人脸追踪(环境准备)

本教程将详细指导你在笔记本上安装Ubuntu 20.04系统,并配置Nvidia驱动、CUDA、PyTorch和ROS环境,为Python实现人脸追踪项目做好准备。整个过程适合小白,我们会分享常见踩坑点,确保你顺利搭建开发环境。

一、系统准备与更新

首先,确保你的笔记本已安装Ubuntu 20.04。打开终端(Ctrl+Alt+T),运行以下命令更新系统包:

    sudo apt updatesudo apt upgrade -y  

这能避免后续安装中的依赖问题。接下来,我们需要安装Nvidia驱动,这是支持RTX 5060显卡的关键。

二、安装Nvidia RTX 5060驱动

Ubuntu 20.04默认使用开源驱动,但为了GPU性能,必须安装官方驱动。先添加驱动仓库并安装:

    sudo add-apt-repository ppa:graphics-drivers/ppasudo apt updatesudo apt install nvidia-driver-510 -y  # 根据RTX 5060兼容性选择驱动版本,510为示例  

安装后重启系统:sudo reboot。重启后,在终端运行nvidia-smi,如果显示GPU信息,则驱动安装成功。这是CUDA安装的基础。

踩坑指南:笔记本Ubuntu 20.04环境搭建 Ubuntu 20.04  Nvidia驱动 CUDA安装 PyTorch环境 第1张

三、安装CUDA工具包

CUDA是Nvidia的并行计算平台,为PyTorch提供加速。访问Nvidia官网下载CUDA 11.7(兼容RTX 5060和Ubuntu 20.04),或使用终端命令:

    wget https://developer.download.nvidia.com/compute/cuda/11.7.0/local_installers/cuda_11.7.0_515.43.04_linux.runsudo sh cuda_11.7.0_515.43.04_linux.run  

安装过程中,确保选择不安装驱动(因已安装)。完成后,添加环境变量到~/.bashrc

    echo "export PATH=/usr/local/cuda-11.7/bin:$PATH" >> ~/.bashrcecho "export LD_LIBRARY_PATH=/usr/local/cuda-11.7/lib64:$LD_LIBRARY_PATH" >> ~/.bashrcsource ~/.bashrc  

验证CUDA安装:运行nvcc --version,应显示版本信息。

四、配置PyTorch环境

PyTorch是深度学习框架,我们将创建Python虚拟环境来管理依赖。安装Python3和pip(如果未安装):

    sudo apt install python3-pip python3-venv -ypython3 -m venv pytorch_env  # 创建虚拟环境source pytorch_env/bin/activate  # 激活环境  

在虚拟环境中,安装PyTorch(支持CUDA 11.7):

    pip install torch torchvision torchaudio --extra-index-url https://download.pytorch.org/whl/cu117  

验证PyTorch环境:运行Python并输入import torch; print(torch.cuda.is_available()),应返回True,表示GPU可用。

五、安装ROS(Robot Operating System)

ROS用于机器人开发,在人脸追踪中可处理传感器数据。安装ROS Noetic(Ubuntu 20.04版本):

    sudo sh -c "echo "deb http://packages.ros.org/ros/ubuntu $(lsb_release -sc) main" > /etc/apt/sources.list.d/ros-latest.list"sudo apt-key adv --keyserver "hkp://keyserver.ubuntu.com:80" --recv-key C1CF6E31E6BADE8868B172B4F42ED6FBAB17C654sudo apt updatesudo apt install ros-noetic-desktop-full -y  

初始化ROS并设置环境:

    echo "source /opt/ros/noetic/setup.bash" >> ~/.bashrcsource ~/.bashrcsudo rosdep initrosdep update  

现在,你的Ubuntu 20.04系统已准备好驱动、CUDA、PyTorch和ROS,可以开始Python人脸追踪项目了。

六、踩坑与常见问题

  • 驱动冲突:如果Nvidia驱动安装失败,先卸载旧驱动:sudo apt purge nvidia*,再重新安装。
  • CUDA版本不匹配:确保CUDA与PyTorch兼容,参考PyTorch官网选择版本。
  • 虚拟环境问题:如果PyTorch无法识别GPU,检查CUDA路径和虚拟环境是否激活。
  • ROS依赖错误:运行rosdep install --from-paths src --ignore-src -r -y解决缺失包。

遵循本教程,你已成功搭建开发环境。下一步可编写Python代码,结合OpenCV和PyTorch实现人脸追踪。如果有问题,欢迎在评论区讨论!