Mohammad Shafiei
Results
12
comments of
Mohammad Shafiei
@hawkinsp I am actually building jaxlib from source and passing the cuda specifications as below, ``` python ./build/build.py \ --enable_cuda \ --cuda_path='/usr/local/cuda-11.5' \ --cudnn_path='/usr/local/cuda-11.5' \ --cuda_version='11.5' \ --cudnn_version='8.3.0' \ --cuda_compute_capabilities...
I am also looking for the same information.