docker docker run --gpus all -it --rm nvcr.io/nvidia/pytorch:23.10-py3 install NVIDIA/TransformerEngine pip install git+https://github.com/NVIDIA/TransformerEngine.git@stable