安装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