如何在Google Colab中获得分配的GPU规格

Question

我正在使用Google Colab进行深度学习，并且我知道他们会为用户随机分配GPU。我希望能够查看在任何给定会话中分配给我的GPU。是否可以在Google Colab笔记本中执行此操作？

请注意，如果有帮助，我正在使用Tensorflow。

Answer 1

由于您可以在colab中运行bash命令，因此只需运行model = SARIMAX(df_diff, order=(0,0,1), seasonal_order = (0,1,1,12)) results = model.fit() forecast = results.get_prediction(start=-10) mean_forecast = forecast.predicted_mean ：

Answer 2

这使阅读更容易

!nvidia-smi -L

Answer 3

在协作中运行这两个命令

CUDA：让我们检查一下Nvidia CUDA驱动程序是否已经预安装以及它是哪个版本。

!/usr/local/cuda/bin/nvcc --version

!nvidia-smi