You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
|
7 months ago | |
---|---|---|
.gitignore | 7 months ago | |
Dockerfile | 7 months ago | |
LICENSE | 7 months ago | |
README.md | 7 months ago | |
docker-compose.yaml | 7 months ago | |
main.py | 7 months ago |
README.md
Stable Diffusion Example
Install Docker
curl https://get.docker.com | sh
distribution=$(. /etc/os-release;echo $ID$VERSION_ID)
&& curl -s -L https://nvidia.github.io/nvidia-docker/gpgkey | sudo apt-key add -
&& curl -s -L https://nvidia.github.io/nvidia-docker/$distribution/nvidia-docker.list | sudo tee /etc/apt/sources.list.d/nvidia-docker.list
sudo apt-get update
sudo apt-get install -y nvidia-docker2 docker-compose
Run
sudo docker-compose up --build
https://huggingface.co/blog/stable_diffusion
https://discuss.pytorch.org/t/need-help-trouble-with-cuda-capability-sm-86/120235/38