If you have trouble running on Runpod.....

[yushan777]

If anyone is having trouble running this on Runpod. The only docker template from runpod that seems to work is runpod/pytorch:3.10-1.13.1-116 If you don't see it in the list, just duplicate the existing pytorch 2.0 one, and paste runpod/pytorch:3.10-1.13.1-116 into the field named "Container Image" (and rename the Template name)

For VAST.AI, I have tested following pytorch images and they all work

pytorch/pytorch:2.0.0-cuda11.7-cudnn8-runtime
pytorch/pytorch:1.13.1-cuda11.6-cudnn8-runtime
pytorch/pytorch:1.13.0-cuda11.6-cudnn8-runtime
pytorch/pytorch:1.12.1-cuda11.3-cudnn8-runtime
pytorch/pytorch:1.12.0-cuda11.3-cudnn8-runtime
pytorch/pytorch:1.11.0-cuda11.3-cudnn8-runtime
pytorch/pytorch:1.10.0-cuda11.3-cudnn8-runtime

Note* The original dreambooth_runpod_joepenna.ipynb has been renamed to dreambooth_runpod_vastai_joepenna_obsolete.ipynband has been superseded by a more user-friendlydreambooth_joepenna.ipynb