GCP DL映像:基本m36(含CUDA 10.0)。如何在PyTorch中启用GPU?

我正在尝试在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?

bonwemeters 回答:GCP DL映像:基本m36(含CUDA 10.0)。如何在PyTorch中启用GPU?

暂时没有好的解决方案,如果你有好的解决方案,请发邮件至:iooj@foxmail.com
本文链接:https://www.f2er.com/3152606.html

大家都在问