入门Anaconda cuda yolov5 安装排坑

主要参考了  这篇博客  的流程,但是期间还是遇到不少问题

首先pytorch的安装,除了pytorch 还有torchvision torchaudio,对版本的匹配要求高,可以先试试pytorch首页的安装方式,如果不行,就去该网址去安装老版本https://pytorch.org/get-started/previous-versions/

去英伟达开发者网站查阅显卡驱动对应CUDA toolkit版本 并下载

https://developer.nvidia.com/cuda-toolkit-archive

安装pytorch过程中,可以添加清华源

conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/free/
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main/
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/pytorch/
conda config --set show_channel_urls yes

如果用清华源和阿里源,也出现版本不匹配,安装失败的问题,最后是挂梯子下的

如果安装成功,执行下面的的命令都会顺利。。。

import torch
print(torch.cuda.is_available())
print(torch.backends.cudnn.is_available())
print(torch.cuda_version)
print(torch.backends.cudnn.version())

 pycharm建立工程时,一定要选择安装了pytorch环境的interpreter,如果是base环境,就是anaconda的目录下的python.exe,否则就是对应环境下的python.exe

怎么看加对了,右下角点进去interpreter setting里面,点进来中间的框会出现很多package,包含了要用的pytorch等包

入门Anaconda cuda yolov5 安装排坑_第1张图片

imglabeling 安装出错

一般都介绍pip install的方法,但安装之后再次出现版本不匹配的问题安装不了

去https://github.com/heartexlabs/labelImg

下载代码,然后按照后面的windows+anaconda方案安装完成

如直接使用yolo 里面自动数据集,可能会出现,

raise ImportError("Failed to initialize: {0}".format(exc)) from exc
ImportError: Failed to initialize: Bad git executable.
The git executable must be specified in one of the following ways:
    - be included in your $PATH
    - be set via $GIT_PYTHON_GIT_EXECUTABLE
    - explicitly set via git.refresh()

意思是git没有加入环境变量

在import os 后面加入一句

os.environ["GIT_PYTHON_REFRESH"] = "quiet"

 yolov5成功做了一次网络摄像头识别,然后同时安装了其他包,就再也不能跑了,提示OpenCV 需要rebuild

报错里面有这么一段:

Rebuild the library with Windows, GTK+ 2.x or Carbon support. If you are on Ubuntu or Debian, instal

解决方法:

pip install opencv-contrib-python -i https://pypi.tuna.tsinghua.edu.cn/simple

你可能感兴趣的:(机器学习,python,人工智能,深度学习)