skip to Main Content

Ubuntu – How to install objax with GPU support?

I have followed the objax documentation to install the library with GPU support: https://objax.readthedocs.io/en/stable/installation_setup.html i.e. pip install --upgrade objax CUDA_VERSION=11.6 pip install -f https://storage.googleapis.com/jax-releases/jax_releases.html jaxlib==`python3 -c 'import jaxlib; print(jaxlib.__version__)'`+cuda`echo $CUDA_VERSION | sed s:\.::g` However the last step doesn't work. I…

VIEW QUESTION
Back To Top
Search