以上链接借鉴炮哥的
一、环境安装
1.1pytorch的安装
安装Anaconda来管理深度学习的空间,
利用conda create -n pytorch(环境名字) python=3.8 命令来创建base中的一个新的空间
利用conda activate 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
进入PyTorch官网下载下载对应版本的cuda(我的是GTX1650 cuda最高是11.7可以下载低版本的11.3就可以)
conda env list可以查询conda里面的安装列表
1.2Pycharm的安装包配置
在yolov5的文件夹中由一个requirements可以直接在pycharm的终端中安装,但是少数一些安装不成功
打开anaconda prompt
通过conda activate pytorch进入所创建的环境,用pip install 安装包名字来安装
安装时会出现的一些错误
错误1:
pip uninstall protobuf
pip install protobuf==3.19.0
通过这个命令解决
错误2:
调用摄像头出现下述问题
使用Yolov5训练好模型调用电脑自带摄像头时出现问题:TypeError: argument of type “int‘ is not iterable的解决方法
解决方法:找到utils下的datasets.py的280行在url前面加上str,注意这里有两个url(出现这个错误的原因是类型不匹配 把int变成str 好像和youtube还有关)
三、.在Pycharm怎么选择yolov5的环境