❓Question: I cannot run jupyter lab with JAX docker image.

[donhuvy]

I am very tired.

docker run --gpus all -it --name vyjax -p 8888:8888 --shm-size=20g --ulimit memlock=-1 nvcr.io/nvidia/jax:23.08-py3 bash

inside container

pip install --upgrade pip
pip install --upgrade jupyterlab
jupyter lab --allow-root

Then I go to http://127.0.0.1:8888/lab?token=aff98418d414113eaa6b05ea48b842089eff68b6a67f3f09 but did not success .

my log https://gist.github.com/donhuvy/3c655b99ab671136f50bdbbc33a5b154

Please help me run JAX with CUDA. I have a RTX 5090 32 GB in Windows 11 this PC. this is my ultimate goal. Thank you for your time.

[Steboss]

Hey @donhuvy did you solve this problem?