Question

ubntu版本18.04 NVIDIA Smia 440.1.0 CUDA 10.2 GTx 960 张量板2.3.0 tensorboard-plugin-wit 1.7.0 tensorflow-estimator 2.3.0 tensorflow-gpu 2.3.0

我的GPU无法正常工作，或者您可以说它已安装，但是当我运行模型时，它没有分配给GPU 图片在这里

Answer 1

运行以下代码以查看tensorflow是否正在检测您的GPU。如果gpus的数量列为0，则表示未检测到。您需要在systen和cuDNN v7.6.5上安装Cuda 10.1。在这种情况下，如果您正在使用Anaconda，请打开conda提示符并运行conda install cuDNN = 7.6.5。您可能还必须安装CUDA Toolkit 10.1。如果您使用pip安装了tensorflow，则必须下载并安装CUDA Toolkit 10.1并修改环境变量等。我发现最简单的解决方案是使用Conda安装tensorflow，因为它会自动安装工具包和cudnn。如果您正在使用Anaconda，则打开conda提示符并运行conda fistall --upgrade tensorflow

import tensorflow as tf
from tensorflow.python.client import device_lib
print(device_lib.list_local_devices())
print(tf.__version__)
print("Num GPUs Available: ", len(tf.config.experimental.list_physical_devices('GPU')))
tf.test.is_gpu_available()
!python --version

当我训练模型时，“ 1物理GPU，0逻辑GPU”无法正常运行

1 个答案: