Java教程

【YOLO】环境搭建

本文主要是介绍【YOLO】环境搭建，对大家解决编程问题具有一定的参考价值，需要的程序猿们随着小编来一起学习吧！

 目的：安装PyTorch的gpu版，使用YOLOv5的requirements.txt创建新的conda环境。

 

1. 查看CUDA版本，本机 cuda11.1

nvcc --version

pytorch官网：https://pytorch.org/get-started/locally/

 

2.创建环境

conda create -n yolo pytorch torchvision torchaudio cudatoolkit=11.1 -c pytorch -c conda-forge
pip install -r D:\WorkSpace\deeplearning\yolov5\yolov5\requirements.txt
这篇关于【YOLO】环境搭建的文章就介绍到这儿，希望我们推荐的文章对大家有所帮助，也希望大家多多支持为之网！
您可能喜欢
栏目导航