Windows 10
JAX current last master 432ef31342aa86b7c85959c8afd2b70fc3e53675
Python 3.7.9 wheel
GPU support: CUDA 10.1, cuDNN 7.6.4 The compute capability version: 6.1 (GTX 1080)
Full build arguments used:
python build/build.py --bazel_options="--local_ram_resources=2048" --bazel_startup_options="--output_base=H:/tools/bazel" --bazel_startup_options="--output_user_root=H:\tools\bazel_root" --enable_cuda --cuda_path="C:/Program Files/NVIDIA GPU Computing Toolkit/CUDA/v10.1" --cudnn_path="C:/Program Files/NVIDIA GPU Computing Toolkit/CUDA/v10.1" --cuda_compute_capabilities="6.1" --cuda_version="10.1" --cudnn_version="7.6.4"