大数据及机器学习带 GPU 必须干的第一件事

一、背景

这不仅仅是 Python 语言类进行机器学习解决性能问题的唯一，在使用 Java/Python 进行大数据处理时，有时候也会有遇到需要 GPU 来解决性能的问题。

以作者的使用经验来看，在需要并行计算能力的程序里，CPU 再强壮也比不过 GPU 的加持。

所以 GPU 很重要，但不了解的朋友们，可能一买来爱机，就急急忙忙地开发体验，却不知所以然，有可能是因为少干了几件事！非得要等开发了，才一堆报错，如下所示：

  File "..\torch\cuda\__init__.py", line 239, in _lazy_init
    raise AssertionError("Torch not compiled with CUDA enabled")
AssertionError: Torch not compiled with CUDA enabled

# 又或者是

xFormers can't load C++/CUDA extensions. xFormers was built for:
caught exception 'Torch not compiled with CUDA enabled', memory monitor disabled

上面的图不是唯一的错误，现归类一下，大概不了解的朋友们一共会出现三类：

1. 没有 GPU 却强制写了 GPU 的代码；
2. 有 GPU 但没有安装驱动；
3. 有 GPU 也装了驱动，却没有装正确的库； 

二、安装显卡驱动程序

请自行前往