yolov5环境配置

发布时间 2023-07-17 17:30:45作者: 花YY

基本配置:

操作系统:windows10
IDE:Pycharm
python版本:anaconda Pyhon3.8
pytorch版本:torch 1.10.0
cuda版本:11.3
显卡:RTX 2060 super

①安装pytorch-gpu版本

下载离线安装包,地址:https://download.pytorch.org/whl/torch_stable.html

import torch
torch.cuda.is_available() ## 判断pytorch-gpu是否安装成功

 ②安装其它库

# Base ----------------------------------------
pip install matplotlib
pip install numpy
pip install opencv-python  ##?下载whl文件离线安装。https://www.lfd.uci.edu/~gohlke/pythonlibs/
pip install pillow
pip install pyyaml
pip install requests
pip install scipy
pip install tqdm

# Logging -------------------------------------
pip install tensorboard

# Plotting --------------------------------------
pip install pandas
pip install seaborn

# Export --------------------------------------

# Extras ---------------------------------------
pip install thop  ## ? pip install --upgrade git+https://github.com/Lyken17/pytorch-OpCounter.git
pip install Cython
pip install pycocotools ##conda install pycocotools