为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

相关推荐
石云升3 分钟前
我在企业落地AI这一年,全流程案例拆解,企业落地AI必看!
人工智能·ai编程
MM_MS7 分钟前
Halcon基础知识点及其算子用法
开发语言·人工智能·python·算法·计算机视觉·视觉检测
温柔只给梦中人10 分钟前
深度学习:day03-04
人工智能·深度学习
棒棒的皮皮16 分钟前
【深度学习】YOLO 模型部署全攻略(本地 / 嵌入式 / 移动端)
人工智能·深度学习·yolo·计算机视觉
a程序小傲17 分钟前
小红书Java面试被问:TCC事务的悬挂、空回滚问题解决方案
java·开发语言·人工智能·后端·python·面试·职场和发展
竹君子1 小时前
AIDC知识库(3)英伟达Rubin 架构对未来AIDC方案的影响初探
人工智能
棒棒的皮皮1 小时前
【深度学习】YOLO模型速度优化全攻略(模型 / 推理 / 硬件三层维度)
人工智能·深度学习·yolo·计算机视觉
线束线缆组件品替网1 小时前
Amphenol RF 同轴线缆:高频 RF 系统设计中 VSWR 与损耗控制实践
网络·人工智能·电脑·硬件工程·材料工程
土星云SaturnCloud1 小时前
液冷技术的未来:相变冷却、喷淋冷却等前沿技术探索
服务器·人工智能·ai
悟道心1 小时前
7. 自然语言处理NLP - Bert
人工智能·自然语言处理·bert