PyTorch 从入门到精通
文章平均质量分 80
大数据AI
专注于IT技术分享
展开
-
loss.sum().backward()中对于sum()的理解
PyTorch backward() 进行梯度计算时,只能对标量进行梯度计算。.sum() 函数主要有两个作用,一个是用来求和,一个是用来降维。在深度学习中,损失函数都是标量,所以一般情况下可以直接调用backward()就可以了。原创 2024-04-30 10:39:22 · 574 阅读 · 0 评论 -
入门级深度学习主机组装过程
大小,最用在 RTX3060 12G显存和RTX 4060 16G显存 之间进行了选择。最后买了RTX 4060 16G显存的显卡。显卡大小:显卡有1风扇、2风扇和3风扇之分,风扇越多散热性能越好,但是我的电脑是itx 小机箱,最后买了个2风扇的显卡。除了显卡和电源,其他硬件都是公司电脑原装。买显卡时,都会写建议电源功率。有钱直接上 RTX4090,也不能复用公司的电脑,其他配置跟不上。担心散热问题,换了个好点的风扇。后来发现还不如不换,太能折腾了。进行深度学习,除了看算力外,还需要看。原创 2024-04-03 06:46:30 · 435 阅读 · 0 评论 -
史上最强 PyTorch 2.2 GPU 版最新安装教程
史上最强 PyTorch 2.2 GPU 版最新安装教程原创 2024-04-03 06:36:37 · 2243 阅读 · 2 评论