YOLO环境搭建

Error专业户

已于 2024-06-22 15:22:28 修改

阅读量41

点赞数

分类专栏：其他文章标签： YOLO

于 2023-03-02 14:01:31 首次发布

本文链接：https://blog.csdn.net/qq_23339307/article/details/129295740

版权

其他专栏收录该内容

15 篇文章 0 订阅

订阅专栏

2023.3.2

YOLOv7环境搭建参考(282条消息) YOLOv7（目标检测）入门教程详解---环境安装_yolo安装_螺丝工人的博客-CSDN博客

Anaconda安装：下载地址：Anaconda | The World's Most Popular Data Science Platform

Pycharm安装：下载地址：PyCharm：JetBrains为专业开发者提供的Python IDE

cuda安装：下载地址：CUDA Toolkit Archive | NVIDIA Developer

cuda安装前参看可以安装的版本号，之前安装的可以卸载，可以保留physx，图形驱动，GeForce。

安装完成后进行环境变量设置-路径加进去（一般电脑自动设置）

cudnn下载：CUDA Deep Neural Network (cuDNN) | NVIDIA Developer，下好的文件复制粘贴进cuda安装的文件目录下。


cmd
nvidia-smi     #查看最大安装版本
nvcc -V      #安装完成后查看安装版本

虚拟环境搭建 Pytorch安装：PyTorch 找到对应的版本复制粘贴


进入Anaconda Prompt
conda env list  #查看conda虚拟环境
conda create -n yolo python=3.8    #搭建环境名为yolo python版本为3.8
activate yolo  #激活yolo环境
conda install conda install pytorch torchvision torchaudio pytorch-cuda=11.7 -c pytorch -c nvidia
#根据pytorch版本号进行复制   下载安装即可

下载中断，或者某个包异常，注意不要删除其他包文件以及url.txt，可以将下载好的包放入环境包文件中，并在url.txt中添加该报路径，进入安装即可。也可设置下载时长，设置镜像源网站


pip --default-timeout=1000.0 install requests  #设置超市时间
pip config set global.index-url https://pypi.tuna.tsinghua.edu.cn/simple  #永久使用清华源

查看环境安装效果


cativate yolo
python
import torch
torch.__version__    #查看torch版本号
torch.cuda.is_available  #查看cuda是否可用
torch.cuda.device_count   #查看可用cuda设备
torch.version.cuda     #查看cuda版本

总结：cuda 和pytorch以及python版本对应，确定了无需修改，高版本torch较大的兼容低版本的。