All looking good except that you are using an old version of the transformers library. Try transformers_version=“4.12.3”, pytorch_version=“1.9.1”, and py_version=“py38”. That should work 
Just for reference, you can find the latest version of the DLCs here: Reference
Cheers
Heiko