随笔分类 -  pytorch

pytorch 求导
摘要：import torch a = torch.ones([3, 4],requires_grad=True) b = 3 c = a * b c.backward(torch.ones_like(c)) print(a.grad) 阅读全文
posted @ 2025-02-06 22:18 恰逢其时40 阅读(25) 评论(0) 推荐(0)
pytorch2.0.1 编译
摘要：conda create -n xxx git clone --recursive https://github.com/pytorch/pytorch.git -b v2.5.1 (如果ssh需要秘钥） git submodule sync git submodule update --init 阅读全文
posted @ 2025-01-19 13:59 恰逢其时40 阅读(132) 评论(0) 推荐(0)