2024新代码,没有教程,原论文应该是在Linux环境下运行的
一、环境配置细节
conda create -n 【环境名】 python=3.10
必须安装3.10,因为后续triton只能在python3.10和3.11编译
conda install pytorch torchvision torchaudio pytorch-cuda=11.8 -c pytorch -c nvidia
cuda必须安装11.8版本,因为后续
二、修改默认下载路径
huggingface_hub下载模型默认保存在C盘,空间占据过大
.\envs\amodal\Lib\site-packages\diffusers\utils\constants.py
.\envs\amodal\Lib\site-packages\huggingface_hub\constants.py
两文件中的默认路径进行修改
虽然下载路径修改了,但后续仍然会报错
在环境变量中增加默认路径
三、COCOA标注方式
COCOA注释与COCO注释并非同一种,作者给出的链接需要发邮件申请才可COCOA annotations
在本文档中上传了该注释资源