为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

相关推荐
tanis_207712 小时前
MinerU2.5-Pro 中文 PDF 识别准确率全解:OmniDocBench v1.6 权威基准数据
人工智能·python·pdf
我是发哥哈12 小时前
跨AI模型生成视频的五大维度对比:选型避坑指南
大数据·人工智能·学习·机器学习·chatgpt·音视频
如去13 小时前
第四篇《AI+教育:个性化学习的实现路径与教育公平的再平衡》
人工智能
Elastic 中国社区官方博客13 小时前
Elastic 9.4:Workflows 正式发布、Agent Builder 更新,以及 Prometheus / PromQL 支持
运维·数据库·人工智能·elasticsearch·搜索引擎·信息可视化·prometheus
机器视觉_Explorer13 小时前
【halcon】编程技巧:鼠标擦除
图像处理·人工智能·深度学习·算法·视觉检测
杨航 AI14 小时前
XGBoost · 登录防欺诈示例
人工智能
拖拖76514 小时前
Scaling Laws for Neural Language Models:大模型为什么可以被“规模化预测”?
人工智能
何陋轩14 小时前
Spring AI实战指南:在Java项目中集成大语言模型
人工智能·后端·机器学习
暗夜猎手-大魔王14 小时前
转载--Karpathy 怎么看 AI Agent(三):怎么给 Agent 搭一个真正能用的上下文
人工智能
每日综合14 小时前
UKey Wallet 产品体系:移动端应用、硬件安全设备与助记词备份设备
人工智能