New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Jump to bottom

Update CONTRIBUTING.md #97

Merged

cgarciae merged 1 commit into master from contributing-docs-patch-1

Oct 22, 2020

Collaborator

haruiz commented Sep 26, 2020 •

edited

CONTRIBUTING.md update:

Before of poetry install I added:

poetry config virtualenvs.in-project true --local

After installing all the dependencies Jax doesn't recognize the GPU, so I included:

In order for Jax to recognize your GPU, you will probably have to install this library again using the command below:

PYTHON_VERSION=cp38  
CUDA_VERSION=cuda101  # alternatives: cuda100, cuda101, cuda102, cuda110, check your cuda version
PLATFORM=manylinux2010_x86_64  # alternatives: manylinux2010_x86_64
BASE_URL='https://storage.googleapis.com/jax-releases'
pip install --upgrade $BASE_URL/$CUDA_VERSION/jaxlib-0.1.55-$PYTHON_VERSION-none-$PLATFORM.whl
pip install --upgrade jax


          Update CONTRIBUTING.md

be2fe07

cgarciae merged commit d133f28 into master

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment