gpt4 book ai didi

linux - 在 pytorch 中,cuda.is_availbale(),但每个操作都因内存不足而失败

转载 作者:塔克拉玛干 更新时间:2023-11-03 01:17:41 25 4
gpt4 key购买 nike

我已经在 Ubuntu 16 上使用 K80 GPU 设置了一个新的虚拟机(在 GCP 上)。已关注installation instructions适用于 CUDA 工具包 9.1

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

我在安装中缺少什么?

最佳答案

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

关于linux - 在 pytorch 中,cuda.is_availbale(),但每个操作都因内存不足而失败,我们在Stack Overflow上找到一个类似的问题: https://stackoverflow.com/questions/48486408/

25 4 0
Copyright 2021 - 2024 cfsdn All Rights Reserved 蜀ICP备2022000587号
广告合作:1813099741@qq.com 6ren.com