Giter Club home page Giter Club logo

Comments (2)

caniyabanci76 avatar caniyabanci76 commented on July 1, 2024

try one of the following images instead of latest.

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

from dreambooth-stable-diffusion.

elo0i avatar elo0i commented on July 1, 2024

try one of the following images instead of latest.

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

Ohhh my god thank you soo much, now it's working with "pytorch/pytorch:1.13.1-cuda11.6-cudnn8-runtime" (training started and seems well)

It's strange becouse I can use pytorch:latest if I use this repo the normal way with the dreambooth jupyter notebook but if I use a script based on the notebook with my fork (only modified setup_training.py and download_model.py to NOT work as a form and instead receive the args from my script) i get the out of memory error. Maybe becouse my script consumes more memory and the :1.13.1 consumes less than the :latest ?

from dreambooth-stable-diffusion.

Related Issues (20)

Recommend Projects

  • React photo React

    A declarative, efficient, and flexible JavaScript library for building user interfaces.

  • Vue.js photo Vue.js

    🖖 Vue.js is a progressive, incrementally-adoptable JavaScript framework for building UI on the web.

  • Typescript photo Typescript

    TypeScript is a superset of JavaScript that compiles to clean JavaScript output.

  • TensorFlow photo TensorFlow

    An Open Source Machine Learning Framework for Everyone

  • Django photo Django

    The Web framework for perfectionists with deadlines.

  • D3 photo D3

    Bring data to life with SVG, Canvas and HTML. 📊📈🎉

Recommend Topics

  • javascript

    JavaScript (JS) is a lightweight interpreted programming language with first-class functions.

  • web

    Some thing interesting about web. New door for the world.

  • server

    A server is a program made to process requests and deliver data to clients.

  • Machine learning

    Machine learning is a way of modeling and interpreting data that allows a piece of software to respond intelligently.

  • Game

    Some thing interesting about game, make everyone happy.

Recommend Org

  • Facebook photo Facebook

    We are working to build community through open source technology. NB: members must have two-factor auth.

  • Microsoft photo Microsoft

    Open source projects and samples from Microsoft.

  • Google photo Google

    Google ❤️ Open Source for everyone.

  • D3 photo D3

    Data-Driven Documents codes.