Skip to content

Instantly share code, notes, and snippets.

@ankane
Last active October 29, 2020 23:23
Show Gist options
  • Save ankane/9b2b5fcbd66d6e4ccfeb9d73e529abe7 to your computer and use it in GitHub Desktop.
Save ankane/9b2b5fcbd66d6e4ccfeb9d73e529abe7 to your computer and use it in GitHub Desktop.
# Bootstrap AWS Deep Learning Base AMI (Ubuntu 18.04)
# install Ruby
sudo snap install ruby --classic
# install LibTorch
wget -O libtorch.zip https://download.pytorch.org/libtorch/cu102/libtorch-cxx11-abi-shared-with-deps-1.7.0.zip
unzip libtorch.zip
# download
git clone https://github.com/ankane/torch.rb.git
cd torch.rb
# install dependencies
bundle install
rake compile -- --with-torch-dir=$HOME/libtorch
# test
rake test
# run example
bundle exec ruby examples/mnist/main.rb
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment