Ubuntu20.04下YOLOX环境搭建之———YOLOX安装

安装pytorch

因为我们在上一篇文章中创建了这个Anaconda虚拟环境，接下来要用它创建一个虚拟环境

，虚拟环境的名字可以自己确定，我这里用的是yolox

conda create -n yolox python=3.7

创建成功后激活虚拟环境

conda activate yolox

在创建的环境下安装pytorch，命令如下

conda install pytorch torchvision cudatoolkit=10.2 -c pytorch

注意：cudatoolkit应该和cuda的版本号要一致

编辑~/.bashrc ，设置使用yolox环境下的Python3.7

gedit ~/.bashrc

将下面这句话添加进去

alias python='/home/edem/anaconda3/envs/yolox/bin/python3.7'

注意：python应该是您自己的路径

保存并退出后执行：

source ~/.bashrc

该命令后会回到base环境，再执行conda activate yolox回到自己的虚拟环境

克隆YOLOX并安装

网址：git clone https://github.com/Megvii-BaseDetection/YOLOX

git clone https://github.com/Megvii-BaseDetection/YOLOX.git

在YOLOX目录下执行

python -m pip install -r requirements.txt -i https://pypi.tuna.tsinghua.edu.cn/simple

python setup.py develop

安装apex