Ecosyste.ms: Awesome
An open API service indexing awesome lists of open source software.
https://github.com/conorturner/jax-docker
GPU and CPU docker images for jaxlib
https://github.com/conorturner/jax-docker
Last synced: 4 days ago
JSON representation
GPU and CPU docker images for jaxlib
- Host: GitHub
- URL: https://github.com/conorturner/jax-docker
- Owner: conorturner
- License: mit
- Created: 2020-11-27T13:25:36.000Z (almost 4 years ago)
- Default Branch: main
- Last Pushed: 2020-11-28T07:57:46.000Z (almost 4 years ago)
- Last Synced: 2023-11-07T00:43:43.106Z (about 1 year ago)
- Size: 1.95 KB
- Stars: 0
- Watchers: 1
- Forks: 0
- Open Issues: 0
-
Metadata Files:
- Readme: README.md
- License: LICENSE
Awesome Lists containing this project
README
# jax-docker
GPU and CPU docker images for jaxlib
## Basic Usage
CPU:
```bash
docker run -it conorturner/jax-cpu bash
```
GPU with CUDA 11:
```bash
docker run --runtime nvidia -it conorturner/jax-cuda-11 bash
```