Change the device 0 to 1

showlab / Tune-A-Video

[ICCV 2023] Tune-A-Video: One-Shot Tuning of Image Diffusion Models for Text-to-Video Generation

https://tuneavideo.github.io

Apache License 2.0

4.22k stars 384 forks source link

Closed uzumakinaruto19 closed 1 year ago

uzumakinaruto19 commented 1 year ago

How do I change the device 0 to one so that Pytorch will use device cuda 1

@zhangjiewu ?

archana0297 commented 1 year ago

Facing this issue

zhangjiewu commented 1 year ago

How do I change the device 0 to one so that Pytorch will use device cuda 1

@zhangjiewu ?

@uzumakinaruto19 You may use export CUDA_VISIBLE_DEVICES=GPU_ID to specify a GPU device.

zhangjiewu commented 1 year ago

@archana0297 It seems you've installed an incompatible pytorch version. You may reinstall pytorch following the official guideline.