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