3D机器学习:pytorch入门1

1、pyTorch的好处: 1、 方便的使用GPU加速 生成a,b随机矩阵,进行矩阵乘法matmul。记录:结束时刻减去开始时刻,得到并打印使用时间,并说明是设备:CPU计算的,并说明结果矩阵的大小。 这一次,将设备改成cuda,进行GPU加速,同样记录时间,并且运行两次,因为第一次的计算时间包括初始化,第二次才是真正计算时间。 结果:cuda可用,且使用cuda后,速度从0.23秒变成了0.00
相关文章
相关标签/搜索