TensorFlow-GPU安装问题在具有多个CUDA版本的系统上

Question

我使用

安装了Tensorflow-gpu

python3.6上的 sudo pip3 install tensorflow-gpu

我使用的系统同时安装了CUDA 10和CUDA 9.0。

我已经导出了CUDA 9.0路径，但是import tensorflow仍然给我

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

有什么办法可以强制使用TensorFlow使用CUDA 9.0，因为根据官方文档，使用PIP的默认预编译TensorFlow仅适用于CUDA 9.0。

附加信息：

1. 我不想使用Virtualenv，因为我正在为整个系统安装TensorFlow，以便所有用户都可以使用它。

2. 我过去在与Bazel编译后安装了TensorFlow，但只有我才能使用它。即使将CUDA路径导出到其配置文件之后，其他用户也无法。因此，我这次试图使默认的PIP安装工作。我已经成功卸载了以前的TensorFlow安装。

Answer 1

尝试安装不同版本的TensorFlow，例如1.11.0，即支持CUDA 9

Answer 2

导入tensorflow您的环境应该具有numpy。如果已插入，则使用以下命令安装TensorFlow和TensorFlow-GPU。

activate yourEnvName
conda install tensorflow
conda install tensorflow-gpu