subprocess.CalledProcessError: Command returned non-zero exit status 1
apex
跑swin transformer的时候遇到的问题,描述如下:
subprocess.CalledProcessError: Command ‘[’/home/yons/anaconda3/envs/swin/bin/python’, ‘-u’, ‘main.py’, ‘–local_rank=3’, ‘–eval’, ‘–cfg’, ‘configs/swin_base_patch4_window7_224.yaml’, ‘–resume’, ‘swin_base_patch4_window7_224.pth’, ‘–data-path’, ‘imagenet’, ‘–batch-size’, ‘5’]’ returned non-zero exit status 1.
主要还是apex没装好
我的解决方法:
git clone https://github.com/NVIDIA/apex.git
cd apex
python setup.py install --cuda_ext --cpp_ext (中间加入这句话)
pip install -v --disable-pip-version-check --no-cache-dir --global-option="–cpp_ext" --global-option="–cuda_ext" ./
这样下来应该就可以了