在pytorch，cuda.is_availbale（）中，但是每个操作都因内存不足而失败

Question

我在Ubuntu 16上使用K80 GPU设置了一个新的虚拟机（在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

我在安装中缺少什么？

Answer 1

哦..没关系，我必须重新启动机器才能让驱动程序重新加载。

Answer 2

1. 有时GPU内存被上一操作占用，以释放内存，I​​DE内核（spyder / spyder3）可以重新启动或重新启动PC。
如果您使用的是Ubuntu，请
2. 使用 sudo 命令尝试代码。

希望它也对您有用。