运行时错误：没有可用的 CUDA GPU

Question

我想在我的笔记本电脑上训练一个 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？