ankane / torch.rb

Deep learning for Ruby, powered by LibTorch
Other
724 stars 31 forks source link

Compatibility table: more informations #23

Closed lfuszara1 closed 3 years ago

lfuszara1 commented 3 years ago

Can you add to compatibility table libtorch versions information like availability without cuda (cpu only), with version of cuda (10.2 or 11.1 for 1.8.1 version of libtorch) and in future ROCm compatibility information?

I copied from Dockerfile installation instruction of torch.rb and finally it works on Ubuntu 20.10.

ankane commented 3 years ago

Hey @lfuszara1, Torch.rb should work with the CPU and CUDA versions.