-
Notifications
You must be signed in to change notification settings - Fork 86
New issue
Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.
By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.
Already on GitHub? Sign in to your account
Dolly not using GPU #69
Comments
Sign up for free
to join this conversation on GitHub.
Already have an account?
Sign in to comment
I ran into the issue you hit where the GPU isn't being used by pytorch.
First you need to find out which version on cuda is installed, then you need to install pytorch kind of like this:
sudo pip install torch==1.12.1+cu113 --extra-index-url https://download.pytorch.org/whl/cu113
Change the torch version to what you want and change the cuda version to match what's installed in your environment
The text was updated successfully, but these errors were encountered: