最近需要编译一下DCN这个网络,使用conda
管理依赖库是真的方便,需要的东西往里放,只要版本之间不冲突,对编译代码十分友好。
配置
GPU:RTX 3080
系统: Linux Mint 20.1 Cinnamon 4.8.6
Cuda: 11.1
Pytorch: 1.8.1
运行环境
根据需要的依赖库下载对应的conda环境,这里提供我自己使用的conda 环境下载地址,其实有很多依赖库是用不着的:
https://github.com/Phil-Mao/AnacondaRecipies/tree/master/_environments
Clone之后,创建一个conda环境:
conda env create -n mvs -f _environments/mvs.fixed.yml
激活mvs
环境,conda activate mvs
:
DCNv2-latest
这里我用的是一个比较新的版本:
https://github.com/jinfagang/DCNv2_latest
编译
这里需要指定cuda