測試pytorch 調用gpu 加速矩陣相乘. accelerate matrix multiplication

下面是我機器中的cpu和gpu型號 31.4 GiB Intel® Core™ i7-8700K CPU @ 3.70GHz × 12 GeForce GTX 1080 Ti/PCIe/SSE2 64-bit 代碼會在下面給出 先看下整體的輸出效果 對比了float32 float64 分別用numpy,torch cpu 以及torch gpu 運算矩陣相乘 運行1000次 方陣大小1-500

>>阅读原文<<