tensorflow / tfjs

A WebGL accelerated JavaScript library for training and deploying ML models.

Home Page:https://js.tensorflow.org

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

Issue while using tjfs node gpu in wsl2

sc0rp10n-py opened this issue · comments

I first installed cuda toolkit 12.3 in windows side, and then again installed it in wsl 2 side using the official docs

For both I kept getting this error

Could not load dynamic library 'libcudart.so.11.0'; dlerror: libcudart.so.11.0: cannot open shared object file: No such file or directory

now I also added the LD paths for my version.

So i manually went to that path and saw for myself, that there were these libcudart libcudnn etc dlls there, but the ending number was not matching, so I am guessing the tfjs doesn't suppport latest cuda yet, so can you tell me what version does it support

so i install that cuda toolkit on wsl2 side

@gaikwadrahul8 hi do you have any idea about it?

if you can tell me what cuda version does tfjs 4.17.0 support, because the files it is searching for (the DLL versions) are not there anymore in cuda 12.3

Hi, @sc0rp10n-py

I apologize for the delayed response and as per our official documentation

For GPU support, tfjs-node-gpu@1.2.4 or laterrequires the following NVIDIA® software installed on your system as per our tested build configuration and please make sure that you'll have Python version 3.9 or later in the tfjs-node official documentation it's stating it requires Python 2.7 but that documentation is outdated and our relevant team working on it and we will update that official documentation soon :

Name Version
NVIDIA® GPU drivers >450.x
CUDA® Toolkit 11.2
cuDNN SDK 8.1.0

For Windows WSL2 software requirements are :

The following NVIDIA® software are only required for GPU support.

Note: GPU support on native-Windows is only available for 2.10 or earlier versions, starting in TF 2.11, CUDA build is not supported for Windows. For using TensorFlow GPU on Windows, you will need to build/install TensorFlow in WSL2 or use tensorflow-cpu with TensorFlow-DirectML-Plugin

Thank you for your understanding and patience.

yeah thanks, i was able to figure it out

Hi, @sc0rp10n-py

We're glad to hear you were able to resolve your issue!, please feel free to close this issue if the issue has been resolved. if you need any further assistance please feel free to create new issue and we'll be happy to help you.

Thank your for your understanding and patience.

Hi, @sc0rp10n-py

We haven't got confirmation from your end so now I'm closing this issue, if you need any further assistance please feel free to create new issue and we will be happy to help you with your issue.

Thank you for your understanding and patience.

Are you satisfied with the resolution of your issue?
Yes
No