使用 CPU 代替 Cuda

作者: Steinful32创建于 2020年7月23日更新于 2024年11月24日

尝试使用 CPU 而不是 cuda,通过将 GPU ID 替换为 -1,然后进入 main.py 并将“device”替换为“cpu”,map_location=torch.device(cpu) 我得到以下错误代码

内容来源: vt-vl-lab/3d-photo-inpainting