NVIDIA / TorchFort

An Online Deep Learning Interface for HPC programs on NVIDIA GPUs
https://nvidia.github.io/TorchFort/
Other
154 stars 19 forks source link

Using the library without cuda #16

Closed NicoClinco closed 1 month ago

NicoClinco commented 2 months ago

Hello, i was wondering if it is possibile to have a version of this library that run without the cuda support since i want to use the library on cpu.

Thanks in advance for the answer Nico

azrael417 commented 2 months ago

Hello Nico,

thank you for your interest in TorchFort. We are currently looking at CPU support and while I think that you need to have the CUDA SDK to compile the library, running on CPU is supported if the library is compiled with GNU. But let me check that in detail and get back to you. Sorry for the delay.

Best and thanks Thorsten

azrael417 commented 1 month ago

Hello Nico,

we just merged changes to enable building TorchFort without GPU support. In order to do so, you need to specify TORCHFORT_ENABLE_GPU=0 as make defines (-DTORCHFORT_ENABLE_GPU=0), or alternatively use the CPU only Dockerfile. GPU is otherwise enabled by default. We have tested the most common functionality on CPU but if you run into issues, please let us know.

Best and thanks Thorsten

NicoClinco commented 1 month ago

Hi, thank you so much for your work, i will test the functionalities soon. Best wishes, Nicola

Il giorno gio 5 set 2024 alle ore 09:50 Thorsten Kurth < @.***> ha scritto:

Hello Nico,

we just merged changes to enable building TorchFort without GPU support. In order to do so, you need to specify TORCHFORT_ENABLE_GPU=0 as make defines (-DTORCHFORT_ENABLE_GPU=0), or alternatively use the CPU only Dockerfile. GPU is otherwise enabled by default. We have tested the most common functionality on CPU but if you run into issues, please let us know.

Best and thanks Thorsten

— Reply to this email directly, view it on GitHub https://github.com/NVIDIA/TorchFort/issues/16#issuecomment-2330843899, or unsubscribe https://github.com/notifications/unsubscribe-auth/AWTZBDOLLVW5EQP7PBBKEA3ZVAEKVAVCNFSM6AAAAABNAVQTYCVHI2DSMVQWIX3LMV43OSLTON2WKQ3PNVWWK3TUHMZDGMZQHA2DGOBZHE . You are receiving this because you authored the thread.Message ID: @.***>