google colab 使用本地運行時並且不檢測本地 gpu

Question

Google colab 連接到本地運行時

使用以下步驟配置本地 GPU 設置：

編輯 -> 筆記本設置 -> 硬件加速器 (GPU)

運行時 -> 更改運行時類型 -> 硬件加速器 (GPU)

已連接（本地）

然后使用以下代碼檢查在本地運行時運行的 colab 是否使用本地 GPU

import torch
torch.cuda.is_available()

上面的語句返回 False

注意：本地計算機系統使用 NVDIA GeForce 940M GPU

Answer 1

根據Vulkan 后端的Pytorch 教程，

默認情況下，Pytorch 中不包含 Vulkan 后端。

為了將 PyTorch 與 Vulkan 后端一起使用，您需要使用其他設置從源代碼構建它。 也許你最好在本地而不是 Colab 這樣做。

Pytorch 已經支持 CUDA 和其他 GPU 加速器，因此除非您打算將其用於項目推理，否則最好按原樣使用。

如果你打算使用 Vulkan 后端，這個 github 項目有一個 DockerFile 和你需要構建 Pytorch 的所有腳本以及 Pytorch-Android .aar 文件

Github Pytorch Android 示例