当前位置:   article > 正文

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

linux conda 安装pytorch+yolov5

一、安装pytorch(yolo v5匹配)

1.新建环境 。

conda create -n <name> python==3.8

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

conda install pytorch torchvision torchaudio cudatoolkit=11.3 -c pytorch

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:

  1. https://docs.ultralytics.com/tutorials/train-custom-datasets/
  2. 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/

声明:本文内容由网友自发贡献,不代表【wpsshop博客】立场,版权归原作者所有,本站不承担相应法律责任。如您发现有侵权的内容,请联系我们。转载请注明出处:https://www.wpsshop.cn/w/IT小白/article/detail/385856
推荐阅读
相关标签
  

闽ICP备14008679号