【问题标题】:RuntimeError: No CUDA GPUs are availableRuntimeError:没有可用的 CUDA GPU
【发布时间】:2021-08-09 07:44:28
【问题描述】:

我想在我的笔记本电脑上训练一个 gpt2 模型,里面有一个 GPU,我的操作系统是 windows,但我在 python 中总是遇到这个错误:

torch._C._cuda_init()
RuntimeError: No CUDA GPUs are available

当我尝试在 python 控制台中检查 GPU 的可用性时,我得到了正确的结果:

import torch
torch.cuda.is_available()
Out[4]: True

但我无法得到版本

nvcc version 
#or  nvcc --version
NameError: name 'nvcc' is not defined

我用这个命令安装CUDA

conda install pytorch torchvision torchaudio cudatoolkit=10.2 -c pytorch  

如何使 GPU 可用于 python?

【问题讨论】:

  • 你安装了什么版本的pytorch CUDA? nvcc version 的输出是什么?
  • 我使用这个命令安装 CUDA { conda install pytorch torchvision torchaudio cudatoolkit=10.2 -c pytorch}
  • NameError: name 'nvcc' is not defined
  • 您能否编辑问题,以便我提出的所有问题都在问题中?您的问题所有这些信息。
  • 我该如何解决?

标签: python anaconda pytorch gpu


【解决方案1】:

确保您已安装 CUDA。对我来说,我是从 NVIDIA 目录安装的:https://developer.nvidia.com/cuda-downloads

【讨论】:

  • 不,这不能解决问题。 anaconda 命令安装所有必要的运行时组件。不需要单独的 CUDA 工具包,并且希望进一步破坏或使事情复杂化
  • 我在 Windows 中安装了 cuda,然后删除了工具,但在这两种情况下 gpu 在 python 中都不起作用
  • 我需要按顺序激活 GPU 并在我的笔记本电脑上使用它。
猜你喜欢
  • 1970-01-01
  • 1970-01-01
  • 2022-11-02
  • 2011-04-14
  • 2020-05-05
  • 1970-01-01
  • 2021-02-15
  • 1970-01-01
  • 2021-07-12
相关资源
最近更新 更多