代码修改:
DECA\decalib\datasets\detectors.py:
DECA\decalib\datasets\datasets.py:
DECA\demos\demo_reconstruct.py:
DECA\demos\demo_transfer.py:
DECA\demos\demo_teaser.py:
修改Model载入参数
DECA\decalib\deca.py:
DECA\decalib\utils\config.py:
调用时候可以使用👇:
python demos/demo_reconstruct.py --device cuda:5
这个5可以换成其他的GPU ,0对应第一块GPU,1对应第二块GPU......
(可使用export CUDA_VISIBLE_DEVICES=0,1,2,3,4,5,6,7 环境变量用于设置可见的GPU设备列表,以控制哪些GPU设备可以被程序使用。export
是一个用于设置环境变量的shell命令,可以将一个或多个变量添加到当前shell环境中。)