Install Pytorch on AWS Glue

install_pytorch_python27.py

import os
import site
from setuptools.command import easy_install
install_path = os.environ['GLUE_INSTALLATION']
easy_install.main( ["--install-dir", install_path, "torch"] )
reload(site)


import torch
print(torch.__version__)

install_pytorch_python36.py

import os
import site

# reload() has been moved to importlib in Python 3.4
# https://docs.python.org/3/whatsnew/3.4.html#importlib
from importlib import reload

from setuptools.command import easy_install
install_path = os.environ['GLUE_INSTALLATION']
easy_install.main( ["--install-dir", install_path, "torch"] )
reload(site)


import torch
print(torch.__version__)