开发者
资源
我要评分
获取效率
正确性
完整性
易理解
在线提单
论坛求助

安装验证

最小启动示例(运行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张量计算结果正确。