Ubuntu系统使用conda配置yolov5环境

1. 使用官方安装包安装 Conda

前往官方网站（Miniconda — miniconda documentation）下载对应的安装包，根据自己的系统和Python版本选择相应的版本进行下载。下载完成后，打开终端，进入安装包所在目录，执行以下命令：

之后去下载目录执行下面指令：

sh Miniconda3-latest-Linux-x86_64.sh

安装完了之后，打开命令窗口，前面会出现一个(base)表示安装成功进入了conda，(如果不想在conda环境中，输入conda deactivate即可退出。)

2.YoLov5模型目标识别算法模型下载

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

也可以直接手动去选择版本下载

Releases · ultralytics/yolov5 · GitHub

3.安装YoLov5

之后打开命令窗口，进入到你的下载目录中，然后按照顺序输入如下指令创建YoLov5环境，

conda create -n yolov5 python=3.8

conda activate yolov5

cuda的版本号根据自己的情况自行更改，我用的是11.4

conda install pytorch torchvision torchaudio pytorch-cuda=11.4 -c pytorch -c nvidia

pip install -r requirements.txt -i https://pypi.tuna.tsinghua.edu.cn/simple

pip install apex -i https://pypi.tuna.tsinghua.edu.cn/simple

如果安装成功，在环境中pip list即可查看安装的torch、torchvision、torchaudio版本

或者可以执行以下命令查看torch的版本

python

import torch

print(torch.version)

使用torch查看cuda、cudnn版本

import torch

print(torch.version)

print(torch.version.cuda)

print(torch.backends.cudnn.version())

环境安装完毕！