virtualenv中的tensorflow没有找到gpu设备

Question

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

当我在tensorflow中列出本地设备时，

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

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

但是，如果我启动virtualenv并在其上安装tensorflow，

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设备。这里缺少什么？

我怀疑tensorflow没有找到CUDA库，但我不知道如何修复它。

Answer 1

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