具有多个 cuda 版本的系统上的 tensorflow-gpu 安装问题

Question

I installed tensorflow-gpu using

sudo pip3 install tensorflow-gpu on python3.6

The system I am using has both cuda 10 and cuda 9.0 installed on it.

I have exported the cuda 9.0 paths, but import tensorflow still gives me

ImportError: libcublas.so.10.0: cannot open shared object file: No such file or directory

Is there any way I can force tensorflow to use cuda 9.0 because the default pre-compiled tensorflow using pip only works with cuda 9.0 according to the official documentation.

Additional info:

1. I do not want to use a virtualenv because I am installing tensorflow for the entire system so that all users can use it.

2. I have in the past installed tensorflow after compiling with bazel, but only I was able to use it. Other users could not, even after exporting the cuda paths to their profiles. So, I am trying to make the default pip installation work this time. I have uninstalled the previous tensorflow installation successfully.

Answer 1

尝试安装不同版本的 tensorflow，如 1.11.0，即支持 cuda 9 的版本

Answer 2

To import tensorflow your environment should have numpy.So check numpy is installed or not using import numpy? If it is intalled then install tensorflow and tensorflow-gpu using following commands.

activate yourEnvName
conda install tensorflow
conda install tensorflow-gpu