我在Ubuntu16上用K80GPU建立了一个新的虚拟机(在GCP上)。
对于CUDA工具包9.1,请遵循installation instructions。
NVIDIA SMI返回正确:
NVIDIA-SMI 390.12驱动程序版本:390.12
…
在pytorch中,cuda.is_available()返回true,
但任何行动都失败了:
torch.randn(10).cuda() ==>
RuntimeError: cuda runtime error (2) : out of memory at /opt/conda/conda-
bld/pytorch_1512378422383/work/torch/lib/THC/generic/THCStorage.cu:58
安装中缺少什么?
最佳答案
哦。。没关系,我得重新启动机器,让司机们重新装载。