How do I specify a graphics card for inference?

Open luoan7248 opened this issue 1 year ago • 3 comments

How do I specify a graphics card for inference? Setting os.environ["CUDA_VISIBLE_DEVICES"] = '6' didn't work.

Jul 16 '24 07:07 luoan7248

you can use CLP

Jul 16 '24 07:07 luoan7248

os.environ["CUDA_VISIBLE_DEVICES"] = '6' works. Please make sure to run this code before import torch and import diffsynth.

Jul 16 '24 08:07 Artiprocher