本文针对搭载NVIDIA RTX5060显卡的笔记本,详细记录在Ubuntu20.04系统上搭建人脸追踪项目所需的全部环境,包括NVIDIA驱动、CUDA、PyTorch、ROS以及Python依赖。特别适合小白参考,避开常见坑点。
确保笔记本搭载RTX5060显卡,并已安装Ubuntu20.04(推荐20.04.6 LTS)。更新系统:sudo apt update && sudo apt upgrade
RTX5060需要驱动版本≥535。推荐使用官方PPA安装:sudo add-apt-repository ppa:graphics-drivers/ppa && sudo apt update,然后安装推荐驱动:sudo ubuntu-drivers autoinstall 或手动选择:sudo apt install nvidia-driver-545。重启后运行nvidia-smi验证。
从NVIDIA官网下载CUDA 12.1(与PyTorch兼容)。选择runfile或deb安装。安装时不要安装驱动,仅安装CUDA toolkit。添加环境变量:export PATH=/usr/local/cuda-12.1/bin:$PATH 和 export LD_LIBRARY_PATH=/usr/local/cuda-12.1/lib64:$LD_LIBRARY_PATH。验证:nvcc -V。
从NVIDIA开发者网站下载cuDNN for CUDA 12.x,解压并复制到cuda目录。
安装PyTorch:推荐使用pip安装CUDA 12.1版本:pip install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu121。验证CUDA可用:python -c "import torch; print(torch.cuda.is_available())",应输出True。
Ubuntu20.04对应ROS Noetic。按官方步骤: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 update,安装sudo apt install ros-noetic-desktop-full。初始化rosdep:sudo rosdep init && rosdep update。设置环境变量:echo "source /opt/ros/noetic/setup.bash" >> ~/.bashrc。
安装OpenCV等:pip install opencv-python opencv-contrib-python numpy dlib(dlib可能需要编译,可先安装依赖)。
编写Python脚本使用OpenCV Haar Cascade检测人脸,并调用摄像头实时跟踪。代码示例:
import cv2face_cascade = cv2.CascadeClassifier(cv2.data.haarcascades + "haarcascade_frontalface_default.xml")cap = cv2.VideoCapture(0)while True: ret, frame = cap.read() gray = cv2.cvtColor(frame, cv2.COLOR_BGR2GRAY) faces = face_cascade.detectMultiScale(gray, 1.1, 4) for (x,y,w,h) in faces: cv2.rectangle(frame, (x,y), (x+w,y+h), (255,0,0), 2) cv2.imshow("Face Tracking", frame) if cv2.waitKey(1) & 0xFF == ord("q"): breakcap.release(); cv2.destroyAllWindows() 运行该脚本,应能看到摄像头窗口并框出人脸。
常见问题:驱动安装后黑屏(需禁用nouveau)、CUDA版本不匹配、PyTorch CUDA不可用、ROS依赖冲突等。建议严格按照版本对应关系安装。
关键词:Ubuntu20.04安装NVIDIA驱动、RTX5060 CUDA配置、Pytorch人脸追踪、ROS环境搭建 均在文中详细说明。
本文由主机测评网于2026-02-20发表在主机测评网_免费VPS_免费云服务器_免费独立服务器,如有疑问,请联系我们。
本文链接:https://vpshk.cn/20260226142.html