NVIDIA-SMI 失败,因为它无法与 NVIDIA 驱动程序通信。确保已安装并运行最新的 NVIDIA 驱动程序。为什么?

问题描述 投票:0回答:2

我正在尝试在 Google Colab 上运行 stylegan2,但所有文件都在我的云端硬盘上,并避免使用来自 stylegan2 github 的 !git clone。 这是我在特定单元格上的代码:

%tensorflow_version 1.x %cd /content/drive/My Drive/stylegan2-master/ !nvcc test_nvcc.cu -o test_nvcc -run print('Tensorflow version: {}'.format(tf.__version__) ) !nvidia-smi -L print('GPU Identified at: {}'.format(tf.test.gpu_device_name()))
结果:

/内容/驱动器/我的驱动器/stylegan2-master CPU 打招呼。 cudaErrorNoDevice:未检测到支持 CUDA 的设备 张量流版本:1.15.2

NVIDIA-SMI 失败,因为它无法与 NVIDIA 驱动程序通信。确保已安装并运行最新的 NVIDIA 驱动程序。

GPU 识别于:

为什么我无法获取 GPU ? 我是这个领域的新手,所以我可能会错过一些非常简单的东西,但仍然无法在互联网上找到答案。

tensorflow gpu google-colaboratory nvidia
2个回答
22
投票
您必须先在笔记本设置中启用 GPU。

您可以通过单击

Edit > Notebook settings

 并选择 GPU 作为硬件加速器来轻松完成此操作。

应该是这样。


0
投票
对于错误 - /bin/bash:第 1 行:nvidia:找不到命令

请使用此代码:

!nvidia-smi 2>&1 | grep -i“cuda”

!nvidia-smi

它帮助我解决错误。

© www.soinside.com 2019 - 2024. All rights reserved.