如果我的计算能力低于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))
小心评论downvote? – user1735003