由于YOLOv8项目中Pytorch版本需要cuda12.1版本

在官网下载12.1版本的deb包

官网地址

sudo dpkg -i cuda-keyring_1.0-1_all.deb
sudo apt-get update
sudo apt-get -y install cuda

然后需要修改bashrc文件（隐藏文件）

添加

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

保存

参考文章