Stable Diffusion on Jetson AGX Orin and Xavier

Autonomous Machines Jetson & Embedded Systems Jetson Projects
1

Hi all,

I came up with a Dockerfile to run Stable Diffusion inside a contianer on Jetson.

Jetson AGX Orin and Jetson AGX Xavier with big VRAM space make it pretty straight forward to run Stable Diffusion, so have a try if you happen to have the hardware.

“Robots marching down a street in Japan”
grid-0001

1 Like
2

Really cool, thanks for sharing @lekto!

1 Like
3

Hi @lekto . I tried this on my Jetson Nano but got the error bellow with “./docker/run.sh”. I am working on this now but it would be nice if you can point out what is wrong here.

image
image1575×82 53.2 KB

1 Like
4

With only 4GB for RAM, I think it is almost impossible to run Stable Diffusion on Jetson Nano.

5

Yes, thank you for your information @lekto . I just want to run some tests with it.

6

Hi @lekto and @dusty_nv Would it be possible to use the xformers to speedup your image generation on Orin in a Docker container? Xformers · AUTOMATIC1111/stable-diffusion-webui Wiki · GitHub

7

Hi @lphung, if you check @lekto’s Dockerfile, that container image was built against JetPack 5.0:

So you would probably need to rebuild it against the version of JetPack you are on with Nano (JetPack 4.x)

Hi @seflands, I’m not entirely sure if this is applicable or not, but here’s also this optimized version of DALL-E Mini for PyTorch:

1 Like
8

what about Jetson TX2? Does anybody tried to run Stable Diffusion there?