/jax-cuda-docker

This repo contains Dockerfile that can be used to easily run JAX with CUDA support in Docker without JAX and CUDA/cuDNN versions mismatching.

Primary LanguageDockerfile

No issues in this repository yet.