huggingface-transformers/docker/transformers-gpu
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
..
Dockerfile Docker GPU Images: Add NVIDIA/apex to the cuda images with pytorch (#7598) 2020-10-06 15:23:32 +02:00