Skip to content

Instantly share code, notes, and snippets.

@ankane
Last active Oct 29, 2020
Embed
What would you like to do?
# 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