手把手教你:nvidia-smi 显示的 CUDA 版本到底是不是你装的那个|AI工程师踩坑实录·第1篇 - 大雾的小屋

Wait 5 sec.

【摘要】真机踩坑实录第1篇:nvidia-smi右上角的CUDA Version只是驱动上限,不是已装版本。CPU版PyTorch调.cuda()报错,根源是PyPI默认源装的是+cpu轮子。解法很简单:换官方cu源CUDA轮子,不动驱动,实测GPU提速18.4倍。 阅读全文