conda安装pytorch环境(与YOLO v5匹配)(Linux)

一、安装pytorch(yolo v5匹配)

1.新建环境 。

conda create -n  python==3.8

2.在pytorch官网 PyTorch选择对应版本,复制下载命令。

conda install pytorch torchvision torchaudio cudatoolkit=11.3 -c pytorch

conda安装pytorch环境(与YOLO v5匹配)(Linux)_第1张图片

3.将 -c pytorch(pytorch源)改为国内源。

4.浏览这个网页,选择正确系统环境,再将对应的链接复制。

https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/pytorch/

https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/pytorch/linux-64/ 

完整的代码如下:

conda install pytorch torchvision torchaudio cudatoolkit=11.3 -c https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/pytorch/linux-64/

5.下载cuda、cudnn

conda install cudatoolkit=11.3.1
conda install cudn=8.2.1

6.下载依赖

克隆yolov5:

https://docs.ultralytics.com/tutorials/train-custom-datasets/
git clone https://github.com/ultralytics/yolov5 
cd yolov5

下载依赖包:

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

二、目标检测 yolo v5网址参考

https://docs.ultralytics.com/

你可能感兴趣的:(环境配置,python,深度学习,pytorch,目标检测)