为Yolov7环境安装Cuba匹配的Pytorch

1. 查看Cuba版本

方法一

``` nvidia-smi ``` 找到CUDA Version

方法二

Nvidia Control Panel > 系统信息 > 组件 >


2. 安装Cuba匹配版本的PyTorch

URL 复制代码
https://pytorch.org/get-started/locally/

这里使用conda安装

shell 复制代码
conda install pytorch torchvision torchaudio pytorch-cuda=11.6 -c pytorch -c nvidia
shell 复制代码
## pip install torch torchvision torchaudio --extra-index-url https://download.pytorch.org/whl/cu116

3. 异常情况

!在这里插入图片描述(https://img-blog.csdnimg.cn/b7578413e7b542e0b5adb6ccbdef249c.jpeg#pic_center)

Collecting package metadata (current_repodata.json): \ ERROR conda.auxlib.logz:stringify(163): Expecting value: line 1 column 1 (char 0)

复制代码
conda config --show-sources

删除指定文件,再运行上面命令回显空白

重新安装PyTorch

  1. download some necessary tools
shell 复制代码
pip install pycocotools-windows -i https://pypi.tuna.tsinghua.edu.cn/simple
pip install -r requirements.txt
pip install pyqt5
  1. Create a new interpreter based on Anaconda3 python

  2. 安装依赖

shell 复制代码
pip install -r .\requirements.txt
  1. Other Exceptions
    Error

    ImportError: cannot import name 'COMMON_SAFE_ASCII_CHARACTERS' from 'charset_normalizer.constant' (D:\ProgramData\Anaconda3\envs\yolov7\lib\site-packages\charset_normalizer\constant.py)
    pip install chardet

    _pickle.UnpicklingError: STACK_GLOBAL requires str
    删labels文件夹同层的labels.cache

    torch.meshgrid warning
    https://github.com/WongKinYiu/yolov7/pull/1167/commits/109fee31c97ef334a682f06c976ff7c9bf25d87d

    • 复制代码
         yv, xv = torch.meshgrid([torch.arange(ny), torch.arange(nx)])
    • 复制代码
         yv, xv = torch.meshgrid([torch.arange(ny), torch.arange(nx)], indexing='ij')
  2. d

相关推荐
只是甲几秒前
Text2SQL 系列博客 07:4 步选型法(上)- 场景与标准的定义
人工智能·text2sql·nl2sql·ai agent·智能问数
Csvn15 分钟前
AI 应用日志别再 print 了:结构化日志 + request_id 贯穿(O01)
人工智能
空 白II30 分钟前
9.26 大语言模型研究简报:把 Agent 开发做成“数据—训练—Harness”闭环
人工智能·语言模型·自然语言处理
tellmewhoisi36 分钟前
机器学习:集成学习4(XGBoost 的API)
人工智能·机器学习·集成学习
只猪侠GGBond42 分钟前
从“会诊断”到“能验证”:AI FaultLab 的修复验证闭环设计
人工智能
m4Rk_1 小时前
【论文阅读】Agent 记忆机制(81):EMR——用情景记忆避免 Agent 在多步推理中反复绕圈
论文阅读·人工智能·学习·开源·github
Zldaisy3d1 小时前
中科院力学所完成太空增减材制造失重飞行试验
人工智能·制造
帝王铠1 小时前
【AI】一些AI时代的想法闲聊
人工智能
szxinmai主板定制专家1 小时前
RK3588+FPGA异构架构|高速数据采集+边缘AI落地应用全解析
人工智能·嵌入式硬件·fpga开发·架构·zynq
李福春1 小时前
降SpringAI阿里第1掌-亢龙有悔-识势选型
人工智能·架构·腾讯云架构师同盟