search

rapidsai-community / rapidsai-csp-utils

35 stars 63 forks source link

No module named `pynvml` #63

Closed dylanrstewart closed 1 year ago

dylanrstewart commented 2 years ago

When running the rapids-colab.ipynb notebook:

!git clone https://github.com/rapidsai/rapidsai-csp-utils.git
!python rapidsai-csp-utils/colab/env-check.py

Cloning into 'rapidsai-csp-utils'... remote: Enumerating objects: 300, done. remote: Counting objects: 100% (129/129), done. remote: Compressing objects: 100% (74/74), done. remote: Total 300 (delta 74), reused 99 (delta 55), pack-reused 171 Receiving objects: 100% (300/300), 87.58 KiB | 17.52 MiB/s, done. Resolving deltas: 100% (136/136), done. Traceback (most recent call last): File "rapidsai-csp-utils/colab/env-check.py", line 1, in import pynvml ModuleNotFoundError: No module named 'pynvml'

beckernick commented 2 years ago

Could you provide the full set of commands you ran and their output?

cc @taureandyernv , have you run into this?

taureandyernv commented 2 years ago

@drstewart19 , thank you for letting us know. At the moment, we're suggesting everyone opt to try our Sage Maker Studio Live (SMSL) solution over Colab, as SMSL supports the latest stable and nightly versions of RAPIDS and Colab can only support 21.12. This does require you to sign up for SMSL, but it's free.

I'll leave the issue open, as this workaround doesn't address the issue you're facing

lextoumbourou commented 2 years ago

I installed pynvml with pip before running the env-check.py and everything seems to work. 

!pip install pynvml

Takes a long, long time to install though.

taureandyernv commented 1 year ago

We just released pip packages, and it works with Google Colab now that they've updated to Python 3.8 two days ago. I've addressed this issue in #65