search

williamyang1991 / Rerender_A_Video

[SIGGRAPH Asia 2023] Rerender A Video: Zero-Shot Text-Guided Video-to-Video Translation
https://www.mmlab-ntu.com/project/rerender/
Other
2.97k stars 202 forks source link

如何配置GPU #113

Closed ChdDongyang closed 11 months ago

ChdDongyang commented 11 months ago

请问在多卡机器上如何配置使用那一块GPU？我在config文件里没有找到类似gpu_id的配置，在rerender.py中也没有看到相关的调用。

williamyang1991 commented 11 months ago

rerender.py开头添加代码（下面的例子用 GPU 1）

import os
os.environ['CUDA_VISIBLE_DEVICES'] = '1'

或者 调用的时候

CUDA_VISIBLE_DEVICES=1 python rerender.py

我感觉这种问题百度一下就能解决了 https://zhuanlan.zhihu.com/p/385352354

ChdDongyang commented 11 months ago

谢谢！我的理解有点偏差，会继续加强基础的！