ICAMS / python-ace

Other
57 stars 15 forks source link

fitting with GPU #28

Closed jhung12 closed 1 year ago

jhung12 commented 1 year ago

For fitting the model with Tensorflow with GPU

I got this error

2023-02-16 14:32:53.327564: W tensorflow/stream_executor/platform/default/dso_loader.cc:64] Could not load dynamic library 'libcudart.so.11.0'; dlerror: libcudart.so.11.0: cannot open shared object file: No such file or directory

in order to solve it I follow https://www.tensorflow.org/install/pip#install_cuda_with_apt

Hope it helps

jhung12 commented 1 year ago

https://stackoverflow.com/questions/70967651/could-not-load-dynamic-library-libcudart-so-11-0

This also helps