3D機器學習:pytorch入門1

1、pyTorch的好處: 1、 方便的使用GPU加速 生成a,b隨機矩陣,進行矩陣乘法matmul。記錄:結束時刻減去開始時刻,得到並打印使用時間,並說明是設備:CPU計算的,並說明結果矩陣的大小。 這一次,將設備改成cuda,進行GPU加速,同樣記錄時間,並且運行兩次,因爲第一次的計算時間包括初始化,第二次纔是真正計算時間。 結果:cuda可用,且使用cuda後,速度從0.23秒變成了0.00
相關文章
相關標籤/搜索