在进行深度学习时初始化模型,将模型转移到GPU的时候出现了报错(如图):
RuntimeError: Found no NVIDIA driver on your system. Please check that you have an NVIDIA GPU and installed a driver from http://www.nvidia.com/Download/index.aspx
意味着需要gpu来运行,用一下执行代码检查当前的gpu是否启用:
import torch
torch.cuda.is_available() # 返回 True 表示 GPU 可用
【CPU转GPU】
在代码编辑页中找到上方:【代码执行程序】>【更改运行时类型】