YOLOv5环境搭建

  1. 安装conda

1、Conda 是开源的包管理系统和环境管理系统,可以安装软件包的多个版本和依赖,而且方便切换。
可以安装Miniconda,实现安装conda的目的。
下载地址,注意对应的python版本和系统位数。
2、建立一个虚拟环境,例如yolo。

1.1 Python>=3.7.0

这里下载的版本是3.7.9的windows 64位的版本。
下载地址:
https://www.python.org/downloads/windows/

注意 在安装conda时,默认是带python的,也就是 这个单独的python可以不安装

  1. PyTorch>=1.7
conda install pytorch torchvision torchaudio cudatoolkit=11.3 -c pytorch

这意味着conda会从官方源下载指定的包。我们需要把尾端的”-c pytorch”去掉, conda才会从我们设置的国内源去下载。

因为安装了mamba,所以可以用mamba安装

mamba install pytorch torchvision torchaudio cudatoolkit=11.3

下面这个是用conda安装的过程


(envZhao) C:\Users\zzh_m>conda install pytorch torchvision torchaudio cudatoolkit=11.3 -c pytorch
Collecting package metadata (current_repodata.json): done
Solving environment: done

==> WARNING: A newer version of conda exists. <== 6 18 25 28 35 49 50 54 94 116 122 132 135 152 188 216 222 244 255 283 333 337 466 478 658 786 916 current version: 4.11.0 latest 4.12.0 please update conda by running $ -n base -c defaults warning conda.gateways.disk.delete:unlink_or_rename_to_trash(144): could not remove or rename d:\programs\conda411\pkgs\pytorch-1.11.0-py3.7_cuda11.3_cudnn8_0.tar.bz2. this file manually (you may need to reboot free handles) d:\programs\conda411\pkgs\torchvision-0.12.0-py37_cu113.tar.bz2. ## package plan environment location: d:\programs\conda411\envs\envzhao added updated specs: - cudatoolkit="11.3" pytorch torchaudio torchvision the following packages will be downloaded: | build ---------------------------|----------------- blas-1.0 mkl kb https: mirrors.tuna.tsinghua.edu.cn anaconda pkgs main brotlipy-0.7.0 |py37h2bbff1b_1003 ca-certificates-2022.3.29 haa95532_1 certifi-2021.10.8 py37haa95532_2 cffi-1.15.0 py37h2bbff1b_1 charset-normalizer-2.0.4 pyhd3eb1b0_0 cryptography-36.0.0 py37h21b164f_0 1.0 mb cudatoolkit-11.3.1 h59b6b97_2 545.3 freetype-2.10.4 hd328e21_0 idna-3.3 intel-openmp-2021.4.0 haa95532_3556 2.2 jpeg-9d h2bbff1b_0 libpng-1.6.37 h2a8f88b_0 libtiff-4.2.0 hd0e1b90_0 libuv-1.40.0 he774522_0 libwebp-1.2.2 lz4-c-1.9.3 h2bbff1b_1 mkl-2021.4.0 haa95532_640 114.9 mkl-service-2.4.0 py37h2bbff1b_0 mkl_fft-1.3.1 py37h277e83a_0 mkl_random-1.2.2 py37hf11a4ad_0 numpy-1.21.5 py37h7a0a035_1 numpy-base-1.21.5 py37hca35cd5_1 4.4 openssl-1.1.1n 4.8 pillow-9.0.1 py37hdc2b20a_0 pycparser-2.21 pyopenssl-22.0.0 pysocks-1.7.1 py37_1 pytorch-1.11.0 |py3.7_cuda11.3_cudnn8_0 1.23 gb requests-2.27.1 six-1.16.0 pyhd3eb1b0_1 tk-8.6.11 3.3 torchvision-0.12.0 py37_cu113 9.0 typing_extensions-4.1.1 pyh06a4308_0 urllib3-1.26.9 py37haa95532_0 win_inet_pton-1.1.0 xz-5.2.5 h62dcd97_0 zlib-1.2.12 h8cc25b3_2 zstd-1.4.9 h19a0ad4_0 ------------------------------------------------------------ total: 1.90 new installed: blas win-64::blas-1.0-mkl brotlipy win-64::brotlipy-0.7.0-py37h2bbff1b_1003 cffi win-64::cffi-1.15.0-py37h2bbff1b_1 charset-normalizer noarch::charset-normalizer-2.0.4-pyhd3eb1b0_0 cryptography win-64::cryptography-36.0.0-py37h21b164f_0 win-64::cudatoolkit-11.3.1-h59b6b97_2 freetype win-64::freetype-2.10.4-hd328e21_0 idna noarch::idna-3.3-pyhd3eb1b0_0 intel-openmp win-64::intel-openmp-2021.4.0-haa95532_3556 jpeg win-64::jpeg-9d-h2bbff1b_0 libpng win-64::libpng-1.6.37-h2a8f88b_0 libtiff win-64::libtiff-4.2.0-hd0e1b90_0 libuv win-64::libuv-1.40.0-he774522_0 libwebp win-64::libwebp-1.2.2-h2bbff1b_0 lz4-c win-64::lz4-c-1.9.3-h2bbff1b_1 win-64::mkl-2021.4.0-haa95532_640 mkl-service win-64::mkl-service-2.4.0-py37h2bbff1b_0 mkl_fft win-64::mkl_fft-1.3.1-py37h277e83a_0 mkl_random win-64::mkl_random-1.2.2-py37hf11a4ad_0 numpy win-64::numpy-1.21.5-py37h7a0a035_1 numpy-base win-64::numpy-base-1.21.5-py37hca35cd5_1 pillow win-64::pillow-9.0.1-py37hdc2b20a_0 pycparser noarch::pycparser-2.21-pyhd3eb1b0_0 pyopenssl noarch::pyopenssl-22.0.0-pyhd3eb1b0_0 pysocks win-64::pysocks-1.7.1-py37_1 win-64::pytorch-1.11.0-py3.7_cuda11.3_cudnn8_0 pytorch-mutex noarch::pytorch-mutex-1.0-cuda requests noarch::requests-2.27.1-pyhd3eb1b0_0 six noarch::six-1.16.0-pyhd3eb1b0_1 tk win-64::tk-8.6.11-h2bbff1b_0 win-64::torchaudio-0.11.0-py37_cu113 win-64::torchvision-0.12.0-py37_cu113 typing_extensions noarch::typing_extensions-4.1.1-pyh06a4308_0 urllib3 win-64::urllib3-1.26.9-py37haa95532_0 win_inet_pton win-64::win_inet_pton-1.1.0-py37haa95532_0 xz win-64::xz-5.2.5-h62dcd97_0 zlib win-64::zlib-1.2.12-h8cc25b3_2 zstd win-64::zstd-1.4.9-h19a0ad4_0 superseded a higher-priority channel: ca-certificates --> anaconda/pkgs/main
  certifi                                         pkgs/main --> anaconda/pkgs/main
  openssl                                         pkgs/main --> anaconda/pkgs/main

Proceed ([y]/n)?

</==>

3 安装pycharm,然后配置编译环境

默认安装就可以。
在配置python解释器时,把利用mamba安装的pytorch环境目录添加到编译器中。例如这里选择第一个配置选项,虚拟环境配置,然后选择 利用现有环境
把envZhao这个虚拟环境中的python添加上就可以。

YOLOv5环境搭建

在安装依赖环境时,用conda遇到很多不好解决的依赖冲突。
最后采用在conda环境中,用pip安装就可以。

pip --timeout=1000 install -r requirements.txt -i https://mirrors.aliyun.com/pypi/simple/

也就是在conda的虚拟环境yolo下,把这个requirements.txt文件放到用户的根目录下,yolo提示的那个目录就是用户的根目录。如下图所示。

YOLOv5环境搭建

如果在conda中使用pip安装时,出现如下错误:

WARNING: pip is configured with locations that require TLS/SSL, however the ssl module in Python is

可以把如下的路径加到环境变量中即可。

YOLOv5环境搭建

Original: https://blog.csdn.net/zzh_my/article/details/124404961
Author: zzh_my
Title: YOLOv5环境搭建

原创文章受到原创版权保护。转载请注明出处:https://www.johngo689.com/690366/

转载文章受原作者版权保护。转载请注明原作者出处!

(0)

大家都在看

亲爱的 Coder【最近整理,可免费获取】👉 最新必读书单  | 👏 面试题下载  | 🌎 免费的AI知识星球