98fb718577
- 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) |
||
---|---|---|
.. | ||
transformers-cpu | ||
transformers-gpu | ||
transformers-pytorch-cpu | ||
transformers-pytorch-gpu | ||
transformers-pytorch-tpu | ||
transformers-tensorflow-cpu | ||
transformers-tensorflow-gpu |