-
Notifications
You must be signed in to change notification settings - Fork 1.3k
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
runpod version #80
Comments
I'll look into it |
I'm also very interested in this, and am currently exploring it myself. Will try and remember to get back here with any notes/tips/etc. The first issue i've run into is Related refs:
I needed to install
Then I could run the following code to compile a wheel:
Which output into
This was on a RunPod machine using an RTX 3090 Might be able to include this in the repo like what was done in #144 / #149 (assuming I did it right)? The output from
Though unfortunately that truncates the name of the card. Seemingly a simpler/better command to run is
We can tweak the code in the notebook to make use of this as follows: - s = getoutput('nvidia-smi')
+ s = getoutput('nvidia-smi --list-gpus')
if 'T4' in s:
gpu = 'T4'
elif 'P100' in s:
gpu = 'P100'
elif 'V100' in s:
gpu = 'V100'
elif 'A100' in s:
gpu = 'A100'
+ elif 'RTX 3090' in s:
+ gpu = 'RTX 3090'
while True:
try:
- gpu=='T4'or gpu=='P100'or gpu=='V100'or gpu=='A100'
+ gpu=='T4'or gpu=='P100'or gpu=='V100'or gpu=='A100' or gpu=='RTX 3090'
break
except:
pass
print('�[1;31mit seems that your GPU is not supported at the moment')
time.sleep(5) And then a little further down we can install the wheel we just made (which would be changed to the URL in this repo if it's uploaded/included): + elif (gpu=='RTX 3090'):
+ %pip install -q xformers/dist/xformers-0.0.14.dev0-cp310-cp310-linux_x86_64.whl I'm also not really sure why this section is in a When it comes to downloading the model, I see you're basically installing git LGS/etc, but you can actually just make an authenticated
There are a few cells in the notebook (eg. image upload) that import from
There are also some paths configured that make more sense in colab/similar than they do for RunPod, such as:
|
do you have a working jupyter notebook for RunPod? Looking foward to try this faster method on Runpod. |
Not at this stage. I spent a couple days playing around with things to understand the code better last week, ran into some issues, but am fairly sure I figured enough to be able to pull together a simple notebook for it. Just need to find the spare time to do so now. Hoping maybe can look at it again tomorrow all going to plan. |
Got to spend some time looking at some more of the bits and pieces for this today, including a rather lengthy deepdive into getting some xformers compilation stuff figured out, which I somewhat summarised in this followup TLDR:
After that I was able to correctly compile Next time I get a chance to work on this I want to take what I learned there, and figure it back into my main streamlined colab, hopefully getting much closer to having a full working solution. |
Had any luck installing the correct wheel on a runpod? |
@0xdevalias |
No worries. Hope it helps! :) |
Will there be a version to run on a runpod ?
The text was updated successfully, but these errors were encountered: