-
Notifications
You must be signed in to change notification settings - Fork 54
powerai:1.5.4 Tensorflow #274
Comments
I think we weren't shipping CUDA as part of PowerAI back in the 1.5.4 timeframe, so that needed to be downloaded directly from NVIDIA and installed separately as a pre-requisite to PowerAI itself. There's some info in the System Setup section of the 1.5.4 docs: https://www.ibm.com/support/knowledgecenter/SS5SF7_1.5.4/navigation/pai_systemsetup.html Note each version of PowerAI was built against specific CUDA version, and for 1.5.4 that was CUDA 10.0. That may limit the RHEL versions you could install on. The GPU driver version is a bit more flexible--generally newer driver versions should support older CUDA versions. But you'd still need a RHEL OS, CUDA, and driver version that are all supported together. |
Right. Can you share your CUDA installation details? |
I am working on a provided machine so I have very restricted permissions, i.e. no installation. OS is Ubuntu 16.04, CUDA is at 9.2:
nvidia-docker is also installed but I am not sure about the setup:
|
OK, it sounds like your nvidia-docker set up is not working. |
|
Hi,
I am trying to use ibmcom/powerai:1.5.4-all-ubuntu18.04-py3 since it's listed as having TF 1.12.0, which is required for code I am trying to run.
I've also tried
source /opt/DL/tensorflow/bin/install_dependencies
thensource /opt/DL/tensorflow/bin/tensorflow-activate
as recommended from a similar issue, but attemptingimport tensorflow
still fails with ImportError: libcuda.so.1: cannot open shared object file: No such file or directoryThe architecture is ppc64le.
The text was updated successfully, but these errors were encountered: