Is there a way to upgrade Cuda drivers installed?

[callmephilip]

I am getting the following warning:

UserWarning: CUDA initialization: The NVIDIA driver on your system is too old (found version 11070). Please update your GPU driver by downloading and installing a new version from the URL: http://www.nvidia.com/Download/index.aspx Alternatively, go to: https://pytorch.org to install a PyTorch version that has been compiled with your version of the CUDA driver. (Triggered internally at ../c10/cuda/CUDAFunctions.cpp:108.)

[Clivern]

Hey @callmephilip

I started something https://github.com/Clivern/ml-in-a-box/blob/main/ml_in_a_box_22_04.sh to upgrade everything and get it to work on ubuntu 22.04. please feel free to test it.

[callmephilip]

@Clivern thank you! i'll take it for a spin

[callmephilip]

tried 22_04 version and i am getting the following error:

Traceback (most recent call last):
  File "/home/paperspace/.local/bin/jupyter-contrib", line 8, in <module>
    sys.exit(main())
  File "/home/paperspace/.local/lib/python3.10/site-packages/jupyter_core/application.py", line 280, in launch_instance
    super().launch_instance(argv=argv, **kwargs)
  File "/home/paperspace/.local/lib/python3.10/site-packages/traitlets/config/application.py", line 1051, in launch_instance
    app = cls.instance(**kwargs)
  File "/home/paperspace/.local/lib/python3.10/site-packages/traitlets/config/configurable.py", line 575, in instance
    inst = cls(*args, **kwargs)
  File "/home/paperspace/.local/lib/python3.10/site-packages/jupyter_contrib_core/application.py", line 27, in __init__
    self._refresh_subcommands()
  File "/home/paperspace/.local/lib/python3.10/site-packages/jupyter_contrib_core/application.py", line 43, in _refresh_subcommands
    get_subcommands_dict = entrypoint.load()
  File "/usr/lib/python3/dist-packages/pkg_resources/__init__.py", line 2465, in load
    return self.resolve()
  File "/usr/lib/python3/dist-packages/pkg_resources/__init__.py", line 2471, in resolve
    module = __import__(self.module_name, fromlist=['__name__'], level=0)
  File "/home/paperspace/.local/lib/python3.10/site-packages/jupyter_contrib_nbextensions/__init__.py", line 5, in <module>
    import jupyter_nbextensions_configurator
  File "/home/paperspace/.local/lib/python3.10/site-packages/jupyter_nbextensions_configurator/__init__.py", line 18, in <module>
    from notebook.base.handlers import APIHandler, IPythonHandler
ModuleNotFoundError: No module named 'notebook.base'

[Clivern]

Yeah it fails in the final part due to package conflict. Will investigate further!

    python -m pip --no-cache-dir install --upgrade jupyter_contrib_nbextensions jupyterlab-git
    DEBIAN_FRONTEND=noninteractive jupyter contrib nbextension install --user

[joshua-paperspace]

Hi @callmephilip,

We just updated the MLiaB script (Ubuntu22 version) and now has nvidia driver version 535. This image is available in our console for A100s and H100s.

We are looking to update the other MLiaB template for other machine types soon, but in the meantime, you can reference the section below if you want to run an install of your own.

https://github.com/Paperspace/ml-in-a-box/blob/main/ubuntu-22/ml_in_a_box.sh#L156