Bootstrap AWS Deep Learning Base AMI (Ubuntu 18.04)

October 29, 2020 ยท View on GitHub

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