Skip to content

Instantly share code, notes, and snippets.

@mrshu

mrshu/install_pytorch_python27.py

Created August 7, 2019 14:49
Show Gist options
  • Save mrshu/0c02892699c08e5b22c396de040c2144 to your computer and use it in GitHub Desktop.
Save mrshu/0c02892699c08e5b22c396de040c2144 to your computer and use it in GitHub Desktop.
Download ZIP
Install Pytorch on AWS Glue
Raw
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__)
Raw
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__)
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment