TensorFlow in virtualenv 找不到 GPU 设备

Question

我在一台安装了 Python 的 Linux 机器上，我可以看到带有 nvidia-smi 的显卡。

当我在张量流中列出本地设备时，

python3 -c "from tensorflow.python.client import device_lib; print(device_lib.list_local_devices())"

我正确地看到了两个设备，一个 CPU 和一个 GPU。

但是，如果我启动一个虚拟环境并在上面安装张量流，

virtualenv -p /usr/bin/Python3 ~/.virtualenvs/test
source ~/.virtualenvs/test/bin/activate
pip install tensorflow

列出设备的命令，

python3 -c "from tensorflow.python.client import device_lib; print(device_lib.list_local_devices())"

不输出 GPU 设备。这里缺少什么？

我怀疑 CUDA 库没有被张量流找到，但我不确定如何解决它。

Answer 1

要使用 gpu，正确的 python 包是 tensorflow-gpu ，而不是 tensorflow 。