Question

我在安装tensorflow-gpu时尝试了多种资源。我有GeForce GTX 1060 3GB的图形卡。我同时安装了CUDA和CuDNN 10.0，并根据指南设置了路径。但是，我不能使用tensorflow-gpu。每当我尝试检查它是否具有GPU时，都会给我一个错误。你们中是否有人拥有你们所遵循的最佳资源或任何建议？

#1st way of checking
>>> import tensorflow as tf
>>> tf.Session()
2019-11-22 18:29:44.166767: I tensorflow/core/platform/cpu_feature_guard.cc:142] Your CPU supports instructions that this TensorFlow binary was not compiled to use: AVX2
<tensorflow.python.client.session.Session object at 0x000002855CF70888>

#2nd way of checking    
from tensorflow.python.client import device_lib
print(device_lib.list_local_devices())

[name: "/device:CPU:0"
device_type: "CPU"
memory_limit: 268435456
locality {
}
incarnation: 3258227305323620158
]

Answer 1

我遵循了https://www.tensorflow.org/install/pip上的官方安装指南

在安装tensorflow之前，您可能需要卸载所有现有的tensorflow-gpu（CPU）软件包

如果尚未尝试重新启动，请查看是否可以解决问题。

要卸载tensorflow（CPU），请运行：

pip uninstall tensorflow

如果使用pip3：

pip3 uninstall tensorflow

然后安装tensorflow-gpu：

pip install tensorflow-gpu

pip3：

pip3 install tensorflow-gpu

我已经看到人们在卸载tensorflow之前先卸载了protobuf，所以您也可以通过运行以下命令来尝试一下：

pip uninstall protobuf

pip3：

pip3 uninstall protobuf

安装和使用Tensorflow GPU

1 个答案: