search

coiled / examples

Examples using Dask and Coiled
14 stars 3 forks source link

Fix ``pytorch`` environment for package sync #48

Closed jrbourbeau closed 1 month ago

jrbourbeau commented 1 month ago

These are reasonable changes to make anyways, but I noticed this also fixes the package sync problem @scharlottej13 reported offline. @scharlottej13 can you see if the pytorch notebook now works for you?

To be clear, I don't understand why this fixes things, but it seems to. We should figure out the issue folks are running into, but the changes here seem fine regardless.

review-notebook-app[bot] commented 1 month ago

Check out this pull request on  ReviewNB

See visual diffs & provide feedback on Jupyter Notebooks.

Powered by ReviewNB

ntabris commented 1 month ago

As in switching py3.10 to py3.11 fixes this?

jrbourbeau commented 1 month ago

Either that or dropping pynvml (which we automatically add now). I've not tried to reproduce the problem @scharlottej13 reported though on main.

scharlottej13 commented 1 month ago

Thanks @jrbourbeau! Linking to the slack thread too: https://coiledcomputing.slack.com/archives/C064FUMHTSB/p1715887638735399

Confirming this solves my problem! Adding nvidia as a channel worked too.

jrbourbeau commented 1 month ago

Confirming this solves my problem!

🎉 🤷