Jimmy 9a776da515 | ||
---|---|---|
.gitignore | ||
Dockerfile | ||
LICENSE | ||
README.md | ||
docker-compose.yaml | ||
main.py |
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