gorodnitskiy/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.
Dockerfile
No issues in this repository yet.
This repo contains Dockerfile that can be used to easily run JAX with CUDA support in Docker without JAX and CUDA/cuDNN versions mismatching.
Dockerfile
No issues in this repository yet.