Replace environment files with more lightweight versions

[jerrylin96]

Environment files are a bit bloated, replacing them with lightweight versions that do the same thing will help a lot (especially once data_utils.py has been refactored to completely remove the TensorFlow dependency.