☰

Tensorflow 未检测到 GPU - lambdalabs

tag-icon • tag-icon nvidia python gpu tensorflow
Tensorflow 未检测到 GPU - lambdalabs

我尝试在虚拟环境（venv）中启动并运行 TensorFlow GPU：

我使用的 lambdalabs 操作系统是 Ubuntu 20.04.3 LTS。

我有以下python脚本：checkGPY.py：

import tensorflow as tf

if tf.test.gpu_device_name():
    print('Default GPU Device: {}'.format(tf.test.gpu_device_name()))
else:
    print("Please install GPU version of TF")

在 venv 之外，它运行良好。我获得默认 GPU 设备：/device:GPU:0。如果训练一个小型神经网络 (NN)，watch nvidia-smi我发现 GPU 内存在训练期间会增加。因此 GPU 资源用于 NN 训练。

但是如果我运行它，它是在 venv 里面（我在 venv 里面安装了 tensorflow 版本：2.6.0。）

(venv) x@y $ python checkGPU.py

我得到：请安装 GPU 版本的 TF

我还获得以下信息：无法加载动态库“libcudnn.so.8”；dlerror：libcudnn.so.8：无法打开共享对象文件：没有此文件或目录

所以我明白动态库 libcudnn.so.8 无法从 venv 内部访问。

我该如何解决这个问题？

答案1

为了解决这个问题，我按照使用 lambdalabs 在 venv 中使用 Tensorflow

相关内容