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

7

我想在Google Colab上运行stylegan2,但要将所有文件存储在我的Drive中,并避免使用来自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()))

结果如下:

/content/drive/My Drive/stylegan2-master CPU问候你。 cudaErrorNoDevice: 没有检测到CUDA兼容设备。 Tensorflow版本:1.15.2 NVIDIA-SMI失败,因为无法与NVIDIA驱动程序通信。请确保安装并运行最新的NVIDIA驱动程序。

已识别的GPU:

为什么我无法使用GPU? 由于我是该领域的新手,可能会错过一些非常简单的东西,但仍然无法在互联网上找到答案。

1个回答

21

您需要首先在笔记本设置中启用GPU。

您可以通过单击编辑>笔记本设置,然后选择GPU作为硬件加速器来轻松完成此操作。

就这样。


1
抱歉,我完全误解了你的答案。 - talonmies
1
在我的情况下,我已经在多个笔记本上尝试过这个方法,即使重新启动运行时,仍然没有分配GPU或TPU。 - foxtrotuniform6969

网页内容由stack overflow 提供, 点击上面的
可以查看英文原文,
原文链接