安装验证
最小启动示例(运行GPU张量计算示例)。
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 | python3 - <<'PY' import ctypes import paddle print("load_libcuda", ctypes.CDLL("libcuda.so.1")._name) print("paddle_version", paddle.__version__) print("compiled_with_cuda", paddle.device.is_compiled_with_cuda()) print("device_count", paddle.device.cuda.device_count()) paddle.set_device("gpu:0") x = paddle.arange(6, dtype="float32").reshape([2, 3]) y = paddle.ones([3, 2], dtype="float32") z = paddle.matmul(x, y) + 1 paddle.device.synchronize() print("place", z.place) print("result", z.numpy().tolist()) PY |
预期输出如下信息:
1 2 3 4 5 6 | load_libcuda libcuda.so.1 paddle_version 3.3.0.dev20260319 compiled_with_cuda True device_count 1 place Place(gpu:0) result [[4.0, 4.0], [13.0, 13.0]] |
通过标准为:
- import paddle成功。
- compiled_with_cuda为True。
- device_count大于0。
- 张量实际运行在gpu:0。
- GPU张量计算结果正确。
父主题: 安装指南