We strongly suggest using conda (or miniconda) for package management. For this assignment, you will need to install PyTorch. For any other package dependencies, use conda / pip to install as they come up as needed.
Assignment 3 on transformers for course 16824