python-3.x - GCP DL 图像：基础 m36（使用 CUDA 10.0）。如何在 PyTorch 中启用 GPU？

Question

我正在尝试在 GCP 盒子上运行 DL PyTorch 作业。我采取的步骤：

• 创建一个附有 m36 深度学习映像和 K80 的 VM。首次登录时有一个关于 CUDA 10 驱动程序安装的问题，我选择“y”并设置驱动程序。
• 安装 pyenv
• pyenv install 3.7.5
• pyenv virtualenv 3.7.5 myenv
• pyenv activate myenv
• pip install fastai
• 启动 python 控制台并运行：

代码：

import torch
torch.cuda.is_available()

它返回False。我应该怎么做才能在 PyTorch 中启用 GPU？

标签： python-3.xgoogle-cloud-platformpytorchfast-ai