【发布时间】: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