huggingface-transformers/docker
Adrien David-Sivelle 98fb718577
Docker GPU Images: Add NVIDIA/apex to the cuda images with pytorch (#7598)
- Use cuda:10.2 image instead of 10.1 (to address version mismatch
  warning with pytorch)
- Use devel version that is built on the runtime and includes headers
  and development tools (was otherwise failing to build apex)
2020-10-06 15:23:32 +02:00
..
transformers-cpu Adding Docker images for transformers + notebooks (#3051) 2020-03-04 11:45:57 -05:00
transformers-gpu Docker GPU Images: Add NVIDIA/apex to the cuda images with pytorch (#7598) 2020-10-06 15:23:32 +02:00
transformers-pytorch-cpu Adding Docker images for transformers + notebooks (#3051) 2020-03-04 11:45:57 -05:00
transformers-pytorch-gpu Docker GPU Images: Add NVIDIA/apex to the cuda images with pytorch (#7598) 2020-10-06 15:23:32 +02:00
transformers-pytorch-tpu Add setup for TPU CI to run every hour. (#6219) 2020-08-07 11:17:07 -04:00
transformers-tensorflow-cpu Adding Docker images for transformers + notebooks (#3051) 2020-03-04 11:45:57 -05:00
transformers-tensorflow-gpu Adding Docker images for transformers + notebooks (#3051) 2020-03-04 11:45:57 -05:00