一、下载安装Pycharm、Anaconda
注意:安装Anaconda时候要勾选添加Path到系统环境变量中
二、创建Python虚拟环境+几种必要包
在菜单栏找到Anaconda Prompt,打开
1、安装tensorflow GPU版本(仅限NVIDIA显卡,AMD不可以)
1.1创建Python虚拟环境,和电脑先前安装的Python环境独立
conda create -n yolov4_gpu python=3.6
(python=3.6/3.7/3.8都可以)
选择y
1.2激活刚刚创建的虚拟环境
conda activate yolov4_gpu
1.3按顺序安装tensorflow_gpu、cuda、cudnn
pip install tensorflow-gpu==2.3.1 -i https://mirrors.aliyun.com/pypi/simple/
conda install cudatoolkit=10.1
conda install cudnn
1.4查看tensorflow版本
输入python,这里激活我先前搭配好的环境,这也说明Anaconda能创建很多个虚拟环境,相互之间不干扰,互不影响。
import tensorflow as tf
tf.__version__
三、安装yolov4用到的依赖库
退出步骤二1.4的python环境
回到Anaconda Prompt,确保yolov4_gpu处于激活状态
pip install opencv-python -i https://mirrors.aliyun.com/pypi/simple/
pip install Pillow -i https://mirrors.aliyun.com/pypi/simple/
pip install matplotlib -i https://mirrors.aliyun.com/pypi/simple/