search

LaurentMazare / ocaml-torch

OCaml bindings for PyTorch
Apache License 2.0
410 stars 38 forks source link

Support for CUDA 11 #68

Closed ArulselvanMadhavan closed 1 year ago

ArulselvanMadhavan commented 1 year ago

I have been working on porting mini-dalle using ocaml-torch. I just realized that the GPU I have doesn't support CUDA 10.2. Any plans on upgrading this library to CUDA 11.x versions? What would happen if I try to run this library with libtorch compiled for cuda 11.x ?

ArulselvanMadhavan commented 1 year ago

Also, the link to download cuda 10.2 version of libtorch is no longer working. It leads to access denied page

LaurentMazare commented 1 year ago

Cuda 11 should be supported out of the box (none of the code in ocaml-torch is specific to a cuda version). Thanks for pointing out the broken link, it's now fixed and recommends using 11.7 (but it should work with the other libtorch versions available on the main pytorch.org website).

ArulselvanMadhavan commented 1 year ago

Thank you