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

笔记本Ubuntu20.04 + RTX5060驱动 + CUDA + Pytorch + ROS/Python实现人脸追踪全攻略(环境准备篇)

笔记本Ubuntu20.04 + RTX5060驱动 + CUDA + Pytorch + ROS/Python实现人脸追踪全攻略(环境准备篇)

本文详细记录在笔记本上为Ubuntu20.04配置Nvidia RTX5060驱动CUDAPytorchROS环境,为后续人脸追踪项目打好基础,包含踩坑经验,小白也能跟着做。

1. 硬件与系统检查

首先确认你的笔记本显卡是RTX5060,系统为Ubuntu20.04。使用命令 lspci | grep -i nvidia 查看显卡型号。

笔记本Ubuntu20.04 + RTX5060驱动 CUDA Pytorch ROS/Python实现人脸追踪全攻略(环境准备篇) Ubuntu20.04 Nvidia驱动安装  RTX5060驱动配置 CUDA安装教程 Pytorch人脸追踪环境搭建 第1张

2. 安装NVIDIA驱动(RTX5060驱动)

Ubuntu20.04默认驱动可能不支持RTX5060,需手动安装。推荐使用显卡驱动PPA方式:

sudo add-apt-repository ppa:graphics-drivers/ppasudo apt updatesudo apt install nvidia-driver-535  # 根据实际情况选择版本

安装后重启,运行 nvidia-smi 验证。若遇到Secure Boot问题,需在BIOS中关闭或签名。这是踩坑重点,注意选择。

3. 安装CUDA

前往NVIDIA官网下载对应Ubuntu20.04的CUDA 11.x或12.x(需与驱动兼容)。我们使用CUDA 11.8示例:

wget https://developer.download.nvidia.com/compute/cuda/11.8.0/local_installers/cuda_11.8.0_520.61.05_linux.runsudo sh cuda_11.8.0_520.61.05_linux.run

安装时不要选择安装驱动,只安装CUDA toolkit。配置环境变量 ~/.bashrc 添加:

export PATH=/usr/local/cuda-11.8/bin:$PATHexport LD_LIBRARY_PATH=/usr/local/cuda-11.8/lib64:$LD_LIBRARY_PATH

4. 安装Pytorch

根据CUDA版本选择Pytorch。使用conda或pip安装支持CUDA的Pytorch:

pip3 install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu118

验证:python -c "import torch; print(torch.cuda.is_available())" 应为True。

5. 安装ROS Noetic

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 install ros-noetic-desktop-full

初始化 rosdep,配置环境变量。然后安装Python依赖:pip install rospkg

6. 人脸追踪环境初步测试

编写简单Python脚本,使用OpenCV和Pytorch检测人脸,并发布ROS话题。但本文只聚焦环境准备,测试CUDA和ROS是否正常即可。

本文重点围绕Ubuntu20.04 Nvidia驱动安装RTX5060驱动配置CUDA安装教程Pytorch人脸追踪环境搭建四个关键词,希望帮助读者避免踩坑,顺利搭建环境。

最后,欢迎留言交流。