如果我的计算能力低于3.0,我可以导入Tensorflow与GPU

问题描述:

如果您的GPU不具有3.0或更高的计算能力,并且您导入张量流,您将获得什么输出?张量流仍会导入还是什么?如果我的计算能力低于3.0,我可以导入Tensorflow与GPU

pip install tensorflow-gpu仍然应该正常工作。

只要确保仔细阅读输出。

我想象的最糟糕的情况是它不会像优化那样,因为张量流可能会假设您的设备每秒可以执行多少次计算。

安装完成后,打开python环境并运行它。

import tensorflow 

I tensorflow/stream_executor/dso_loader.cc:105] successfully opened CUDA library libcublas.so locally 
I tensorflow/stream_executor/dso_loader.cc:105] successfully opened CUDA library libcudnn.so locally 
I tensorflow/stream_executor/dso_loader.cc:105] successfully opened CUDA library libcufft.so locally 
I tensorflow/stream_executor/dso_loader.cc:105] successfully opened CUDA library libcuda.so.1 locally 
I tensorflow/stream_executor/dso_loader.cc:105] successfully opened CUDA library libcurand.so locally 

您应该看到类似这样的输出。这就是你将会知道你的GPU是否被正确读取的。 (如果它是NVIDIA,我不知道AMD)

要找出哪个设备使用,你可以运行这个

sess = tf.Session(config=tf.ConfigProto(log_device_placement=True)) 

要在GPU上运行,the requirements are

GPU卡CUDA Compute Capability 3.0或更高。

但是,您仍然可以运行tensorflow的纯CPU版本。

+0

小心评论downvote? – user1735003