ubuntu配置yolov5gpu环境

为了在Ubuntu上配置yolov5-gpu环境，你需要按照以下步骤进行操作：

1. 安装NVIDIA驱动：首先，确保你的系统有一个兼容的NVIDIA显卡。然后，使用以下命令安装NVIDIA驱动：

sudo apt-get update
sudo apt-get install nvidia-driver-<版本号>

替换 `<版本号>` 为你所需的NVIDIA驱动版本号。

2. 安装CUDA Toolkit：yolov5-gpu需要CUDA支持。访问NVIDIA官方网站下载适用于你的显卡和操作系统的CUDA Toolkit，并按照官方文档指引进行安装。

3. 安装cuDNN库：cuDNN是NVIDIA深度神经网络库，用于加速深度学习模型的运行。下载适用于你的CUDA版本的cuDNN库，并按照官方文档进行安装。

4. 创建并激活虚拟环境：使用以下命令创建一个Python虚拟环境，并激活它：

python3 -m venv yolov5-env
source yolov5-env/bin/activate

5. 安装依赖项：在激活的虚拟环境中，使用以下命令安装yolov5-gpu所需的依赖项：

pip install --upgrade pip
pip install wheel numpy torch torchvision matplotlib

6. 克隆yolov5-gpu仓库：使用以下命令克隆yolov5-gpu仓库到本地：

git clone https://github.com/ultralytics/yolov5.git

7. 下载预训练权重：进入yolov5目录，并下载预训练权重：

cd yolov5
python3 -c "from utils.google_utils import