{"id":2438,"date":"2024-10-29T09:48:22","date_gmt":"2024-10-29T01:48:22","guid":{"rendered":"https:\/\/sanlangcode.com\/?p=2438"},"modified":"2024-10-29T09:48:22","modified_gmt":"2024-10-29T01:48:22","slug":"pytorch%e4%b8%ad%e4%b8%bb%e8%a6%81%e7%9a%84%e5%ba%93%e5%92%8c%e5%b8%b8%e7%94%a8%e7%9a%84%e6%a8%a1%e5%9d%97%e7%9a%84%e4%b8%bb%e8%a6%81%e5%8a%9f%e8%83%bd","status":"publish","type":"post","link":"https:\/\/sanlangcode.com\/index.php\/2024\/10\/29\/pytorch%e4%b8%ad%e4%b8%bb%e8%a6%81%e7%9a%84%e5%ba%93%e5%92%8c%e5%b8%b8%e7%94%a8%e7%9a%84%e6%a8%a1%e5%9d%97%e7%9a%84%e4%b8%bb%e8%a6%81%e5%8a%9f%e8%83%bd\/","title":{"rendered":"PyTorch\u4e2d\u4e3b\u8981\u7684\u5e93\u548c\u5e38\u7528\u7684\u6a21\u5757\u7684\u4e3b\u8981\u529f\u80fd"},"content":{"rendered":"\n<p>\u5728 PyTorch \u4e2d\uff0c\u4e3b\u8981\u7684\u5e93\u548c\u5de5\u5177\u5305\u6db5\u76d6\u4e86\u6df1\u5ea6\u5b66\u4e60\u7684\u5404\u4e2a\u65b9\u9762\u3002\u4ee5\u4e0b\u662f PyTorch \u5e38\u7528\u7684\u6a21\u5757\u548c\u5b83\u4eec\u7684\u4e3b\u8981\u529f\u80fd\uff1a<\/p>\n\n\n\n<h3 class=\"wp-block-heading\">1. <code>torch<\/code><\/h3>\n\n\n\n<p>\u8fd9\u662f PyTorch \u7684\u6838\u5fc3\u6a21\u5757\uff0c\u63d0\u4f9b\u4e86\u5f20\u91cf\u64cd\u4f5c\u3001\u81ea\u52a8\u6c42\u5bfc\u548c\u57fa\u7840\u6570\u5b66\u8fd0\u7b97\u529f\u80fd\u3002\u5e38\u7528\u529f\u80fd\u5305\u62ec\uff1a<\/p>\n\n\n\n<ul class=\"wp-block-list\"><li><strong>\u5f20\u91cf\u64cd\u4f5c<\/strong>\uff1a\u521b\u5efa\u548c\u64cd\u4f5c\u5f20\u91cf\uff0c\u4f8b\u5982 <code>torch.tensor()<\/code>\u3001<code>torch.zeros()<\/code>\u3001<code>torch.ones()<\/code>\u3002<\/li><li><strong>\u6570\u5b66\u8fd0\u7b97<\/strong>\uff1a\u63d0\u4f9b\u57fa\u7840\u6570\u5b66\u8fd0\u7b97\u5982\u52a0\u51cf\u4e58\u9664\u3001\u77e9\u9635\u4e58\u6cd5\u7b49\u3002<\/li><li><strong>\u968f\u673a\u6570\u751f\u6210<\/strong>\uff1a\u5982 <code>torch.rand()<\/code>\u3001<code>torch.randn()<\/code> \u7528\u4e8e\u751f\u6210\u968f\u673a\u6570\u7684\u5f20\u91cf\u3002<\/li><li><strong>CUDA \u652f\u6301<\/strong>\uff1a\u4f7f\u7528 <code>.cuda()<\/code> \u548c <code>.to()<\/code> \u53ef\u4ee5\u5c06\u5f20\u91cf\u6216\u6a21\u578b\u79fb\u5230 GPU \u4e0a\uff0c\u4ee5\u52a0\u901f\u8fd0\u7b97\u3002<\/li><\/ul>\n\n\n\n<h3 class=\"wp-block-heading\">2. <code>torch.nn<\/code><\/h3>\n\n\n\n<p><code>torch.nn<\/code> \u662f PyTorch \u63d0\u4f9b\u7684\u795e\u7ecf\u7f51\u7edc\u6a21\u5757\uff0c\u7528\u4e8e\u521b\u5efa\u548c\u7ec4\u5408\u5404\u79cd\u795e\u7ecf\u7f51\u7edc\u5c42\uff0c\u6784\u5efa\u6a21\u578b\u3002\u4e3b\u8981\u7ec4\u4ef6\u5305\u62ec\uff1a<\/p>\n\n\n\n<ul class=\"wp-block-list\"><li><strong>\u795e\u7ecf\u7f51\u7edc\u5c42<\/strong>\uff1a\u5982 <code>nn.Linear<\/code>\uff08\u5168\u8fde\u63a5\u5c42\uff09\u3001<code>nn.Conv2d<\/code>\uff08\u4e8c\u7ef4\u5377\u79ef\u5c42\uff09\u3001<code>nn.RNN<\/code>\uff08\u5faa\u73af\u5c42\uff09\u7b49\u3002<\/li><li><strong>\u6fc0\u6d3b\u51fd\u6570<\/strong>\uff1a\u5982 <code>nn.ReLU<\/code>\u3001<code>nn.Sigmoid<\/code> \u7b49\uff0c\u6dfb\u52a0\u975e\u7ebf\u6027\u64cd\u4f5c\uff0c\u4f7f\u7f51\u7edc\u53ef\u4ee5\u5b66\u4e60\u590d\u6742\u7684\u6620\u5c04\u5173\u7cfb\u3002<\/li><li><strong>\u635f\u5931\u51fd\u6570<\/strong>\uff1a\u5982 <code>nn.MSELoss<\/code>\uff08\u5747\u65b9\u8bef\u5dee\u635f\u5931\uff09\u3001<code>nn.CrossEntropyLoss<\/code>\uff08\u4ea4\u53c9\u71b5\u635f\u5931\uff09\u7b49\uff0c\u7528\u4e8e\u8ba1\u7b97\u9884\u6d4b\u548c\u771f\u5b9e\u503c\u7684\u5dee\u5f02\u3002<\/li><li><strong>\u5bb9\u5668\u6a21\u5757<\/strong>\uff1a\u5982 <code>nn.Sequential<\/code>\u3001<code>nn.ModuleList<\/code> \u7b49\uff0c\u7528\u4e8e\u65b9\u4fbf\u5730\u7ec4\u5408\u548c\u5806\u53e0\u5c42\u3002<\/li><\/ul>\n\n\n\n<h3 class=\"wp-block-heading\">3. <code>torch.optim<\/code><\/h3>\n\n\n\n<p><code>torch.optim<\/code> \u6a21\u5757\u5305\u542b\u4e86\u7528\u4e8e\u4f18\u5316\u6a21\u578b\u7684\u5e38\u7528\u7b97\u6cd5\uff0c\u5373\u66f4\u65b0\u6a21\u578b\u53c2\u6570\u7684\u7b56\u7565\u3002\u5b83\u7684\u4e3b\u8981\u7ec4\u4ef6\u662f\u4f18\u5316\u5668\uff1a<\/p>\n\n\n\n<ul class=\"wp-block-list\"><li><strong>\u4f18\u5316\u5668<\/strong>\uff1a\u5982 <code>optim.SGD<\/code>\uff08\u968f\u673a\u68af\u5ea6\u4e0b\u964d\uff09\u3001<code>optim.Adam<\/code>\uff08\u81ea\u9002\u5e94\u77e9\u4f30\u8ba1\uff09\u7b49\u3002\u4f18\u5316\u5668\u7ba1\u7406\u53c2\u6570\u7684\u66f4\u65b0\uff0c\u5e76\u5229\u7528\u68af\u5ea6\u51cf\u5c11\u635f\u5931\u3002<\/li><li><strong>\u53c2\u6570\u8c03\u6574<\/strong>\uff1a\u53ef\u4ee5\u901a\u8fc7 <code>optim.lr_scheduler<\/code> \u4e2d\u7684\u5b66\u4e60\u7387\u8c03\u5ea6\u5668\u52a8\u6001\u8c03\u6574\u5b66\u4e60\u7387\uff0c\u4ee5\u63a7\u5236\u8bad\u7ec3\u7684\u6536\u655b\u901f\u5ea6\u3002<\/li><\/ul>\n\n\n\n<h3 class=\"wp-block-heading\">4. <code>torch.autograd<\/code><\/h3>\n\n\n\n<p><code>torch.autograd<\/code> \u662f PyTorch \u7684\u81ea\u52a8\u6c42\u5bfc\u6a21\u5757\uff0c\u652f\u6301\u81ea\u52a8\u8ba1\u7b97\u68af\u5ea6\uff0c\u7528\u4e8e\u53cd\u5411\u4f20\u64ad\uff1a<\/p>\n\n\n\n<ul class=\"wp-block-list\"><li><strong>\u53cd\u5411\u4f20\u64ad<\/strong>\uff1a\u901a\u8fc7 <code>loss.backward()<\/code> \u6267\u884c\u53cd\u5411\u4f20\u64ad\uff0c\u5e76\u8ba1\u7b97\u68af\u5ea6\u3002<\/li><li><strong>\u8ba1\u7b97\u56fe\uff08Dynamic Computational Graph\uff09<\/strong>\uff1a\u652f\u6301\u52a8\u6001\u56fe\u673a\u5236\uff0c\u6bcf\u6b21\u524d\u5411\u4f20\u64ad\u4f1a\u91cd\u65b0\u6784\u5efa\u8ba1\u7b97\u56fe\uff0c\u4e3a\u53cd\u5411\u4f20\u64ad\u751f\u6210\u9700\u8981\u7684\u8282\u70b9\u4fe1\u606f\u3002<\/li><li><strong>\u68af\u5ea6\u8bb0\u5f55<\/strong>\uff1a\u901a\u8fc7 <code>torch.no_grad()<\/code> \u53ef\u4ee5\u7981\u7528\u81ea\u52a8\u6c42\u5bfc\uff0c\u5e38\u7528\u4e8e\u63a8\u7406\u65f6\uff0c\u51cf\u5c11\u5185\u5b58\u6d88\u8017\u3002<\/li><\/ul>\n\n\n\n<h3 class=\"wp-block-heading\">5. <code>torch.utils.data<\/code><\/h3>\n\n\n\n<p><code>torch.utils.data<\/code> \u63d0\u4f9b\u6570\u636e\u52a0\u8f7d\u548c\u5904\u7406\u529f\u80fd\uff0c\u4fbf\u4e8e\u7ba1\u7406\u6570\u636e\u96c6\u3002\u4e3b\u8981\u5305\u62ec\uff1a<\/p>\n\n\n\n<ul class=\"wp-block-list\"><li><strong>\u6570\u636e\u96c6\uff08Dataset\uff09<\/strong>\uff1a<code>Dataset<\/code> \u662f\u4e00\u4e2a\u62bd\u8c61\u7c7b\uff0c\u7528\u6237\u53ef\u4ee5\u7ee7\u627f\u5b83\u6765\u521b\u5efa\u81ea\u5b9a\u4e49\u6570\u636e\u96c6\u3002PyTorch \u63d0\u4f9b\u4e86\u5e38\u7528\u6570\u636e\u96c6\u5982 <code>torchvision.datasets<\/code>\u3002<\/li><li><strong>\u6570\u636e\u52a0\u8f7d\u5668\uff08DataLoader\uff09<\/strong>\uff1a<code>DataLoader<\/code> \u7528\u4e8e\u6279\u91cf\u52a0\u8f7d\u6570\u636e\uff0c\u5e76\u652f\u6301\u591a\u7ebf\u7a0b\u52a0\u901f\u3002\u4f7f\u7528\u65b9\u6cd5\uff1a<code>DataLoader(dataset, batch_size, shuffle=True)<\/code>\u3002<\/li><li><strong>\u6570\u636e\u91c7\u6837<\/strong>\uff1a\u5982 <code>RandomSampler<\/code>\u3001<code>SubsetRandomSampler<\/code> \u7b49\uff0c\u7528\u4e8e\u5bf9\u6570\u636e\u8fdb\u884c\u91c7\u6837\uff0c\u5e38\u7528\u4e8e\u6570\u636e\u589e\u5e7f\u548c\u62c6\u5206\u3002<\/li><\/ul>\n\n\n\n<h3 class=\"wp-block-heading\">6. <code>torchvision<\/code><\/h3>\n\n\n\n<p><code>torchvision<\/code> \u662f PyTorch \u9488\u5bf9\u56fe\u50cf\u4efb\u52a1\u7684\u8f85\u52a9\u5e93\uff0c\u5305\u542b\u4e86\u5e38\u7528\u7684\u6570\u636e\u96c6\u3001\u56fe\u50cf\u8f6c\u6362\u548c\u9884\u8bad\u7ec3\u6a21\u578b\uff1a<\/p>\n\n\n\n<ul class=\"wp-block-list\"><li><strong>\u6570\u636e\u96c6<\/strong>\uff1a\u63d0\u4f9b\u4e86\u5e38\u7528\u7684\u56fe\u50cf\u6570\u636e\u96c6\uff0c\u5982 MNIST\u3001CIFAR\u3001ImageNet \u7b49\u3002<\/li><li><strong>\u56fe\u50cf\u8f6c\u6362<\/strong>\uff1a<code>torchvision.transforms<\/code> \u63d0\u4f9b\u4e86\u5e38\u7528\u7684\u56fe\u50cf\u8f6c\u6362\u64cd\u4f5c\uff0c\u5982 <code>transforms.Resize<\/code>\u3001<code>transforms.ToTensor<\/code> \u7b49\uff0c\u7528\u4e8e\u6570\u636e\u9884\u5904\u7406\u548c\u589e\u5f3a\u3002<\/li><li><strong>\u9884\u8bad\u7ec3\u6a21\u578b<\/strong>\uff1a\u5982 <code>torchvision.models<\/code> \u4e2d\u7684 ResNet\u3001VGG \u7b49\u7ecf\u5178\u6a21\u578b\uff0c\u652f\u6301\u76f4\u63a5\u52a0\u8f7d\u9884\u8bad\u7ec3\u6743\u91cd\uff0c\u65b9\u4fbf\u8fdb\u884c\u8fc1\u79fb\u5b66\u4e60\u3002<\/li><\/ul>\n\n\n\n<h3 class=\"wp-block-heading\">7. <code>torchtext<\/code><\/h3>\n\n\n\n<p><code>torchtext<\/code> \u662f\u4e00\u4e2a\u4e13\u4e3a\u81ea\u7136\u8bed\u8a00\u5904\u7406\uff08NLP\uff09\u8bbe\u8ba1\u7684\u5e93\uff0c\u652f\u6301\u6587\u672c\u6570\u636e\u52a0\u8f7d\u548c\u5904\u7406\uff1a<\/p>\n\n\n\n<ul class=\"wp-block-list\"><li><strong>\u6570\u636e\u96c6<\/strong>\uff1a\u5305\u542b\u5e38\u7528\u7684 NLP \u6570\u636e\u96c6\u5982 IMDB\u3001WikiText \u7b49\u3002<\/li><li><strong>\u8bcd\u6c47\u8868\u6784\u5efa<\/strong>\uff1a\u652f\u6301\u6784\u5efa\u8bcd\u6c47\u8868\uff08Vocabulary\uff09\uff0c\u53ef\u4ee5\u8f7b\u677e\u5bf9\u6587\u672c\u8fdb\u884c\u8bcd\u6c47\u7d22\u5f15\u5316\u3002<\/li><li><strong>\u5d4c\u5165\u5c42<\/strong>\uff1a\u652f\u6301\u9884\u8bad\u7ec3\u7684\u8bcd\u5411\u91cf\u5d4c\u5165\uff08\u5982 GloVe\u3001FastText\uff09\uff0c\u7528\u4e8e\u81ea\u7136\u8bed\u8a00\u6a21\u578b\u3002<\/li><\/ul>\n\n\n\n<h3 class=\"wp-block-heading\">8. <code>torchaudio<\/code><\/h3>\n\n\n\n<p><code>torchaudio<\/code> \u662f\u7528\u4e8e\u97f3\u9891\u5904\u7406\u7684\u5e93\uff0c\u5305\u542b\u97f3\u9891\u6570\u636e\u96c6\u3001\u8f6c\u6362\u5de5\u5177\u548c\u7279\u5f81\u63d0\u53d6\u7b49\uff1a<\/p>\n\n\n\n<ul class=\"wp-block-list\"><li><strong>\u97f3\u9891\u6570\u636e\u96c6<\/strong>\uff1a\u63d0\u4f9b\u5e38\u89c1\u7684\u97f3\u9891\u6570\u636e\u96c6\u5982 YESNO \u548c COMMONVOICE \u7b49\u3002<\/li><li><strong>\u7279\u5f81\u63d0\u53d6<\/strong>\uff1a\u652f\u6301\u5e38\u7528\u7684\u97f3\u9891\u7279\u5f81\u63d0\u53d6\uff0c\u5982 Mel \u9891\u8c31\u3001MFCC \u7b49\uff0c\u7528\u4e8e\u97f3\u9891\u5206\u6790\u548c\u8bc6\u522b\u4efb\u52a1\u3002<\/li><\/ul>\n\n\n\n<h3 class=\"wp-block-heading\">9. <code>torch.distributed<\/code><\/h3>\n\n\n\n<p><code>torch.distributed<\/code> \u662f\u7528\u4e8e\u5206\u5e03\u5f0f\u8bad\u7ec3\u7684\u5e93\uff0c\u9002\u5408\u5927\u89c4\u6a21\u6570\u636e\u548c\u6a21\u578b\u7684\u5206\u5e03\u5f0f\u5904\u7406\uff1a<\/p>\n\n\n\n<ul class=\"wp-block-list\"><li><strong>\u5206\u5e03\u5f0f\u6570\u636e\u5e76\u884c<\/strong>\uff1a\u652f\u6301\u591a GPU \u8bad\u7ec3\uff0c\u591a\u4e2a\u8bbe\u5907\u534f\u540c\u5de5\u4f5c\uff0c\u63d0\u9ad8\u8bad\u7ec3\u901f\u5ea6\u3002<\/li><li><strong>\u5206\u5e03\u5f0f\u901a\u4fe1<\/strong>\uff1a\u63d0\u4f9b\u5206\u5e03\u5f0f\u8ba1\u7b97\u4e2d\u7684\u901a\u4fe1\u63a5\u53e3\uff0c\u7528\u4e8e\u4e0d\u540c\u8282\u70b9\u4e4b\u95f4\u7684\u6570\u636e\u4f20\u9012\u3002<\/li><\/ul>\n\n\n\n<h3 class=\"wp-block-heading\">10. <code>torch.jit<\/code><\/h3>\n\n\n\n<p><code>torch.jit<\/code> \u662f PyTorch \u7684\u811a\u672c\u5316\u548c\u8ffd\u8e2a\u529f\u80fd\u6a21\u5757\uff0c\u7528\u4e8e\u6a21\u578b\u4f18\u5316\u548c\u90e8\u7f72\uff1a<\/p>\n\n\n\n<ul class=\"wp-block-list\"><li><strong>\u811a\u672c\u5316\u6a21\u578b<\/strong>\uff1a\u53ef\u4ee5\u5c06 PyTorch \u52a8\u6001\u6a21\u578b\u811a\u672c\u5316\u4e3a\u9759\u6001\u6a21\u578b\uff0c\u63d0\u9ad8\u6267\u884c\u6548\u7387\u3002<\/li><li><strong>\u8ffd\u8e2a\u6a21\u578b<\/strong>\uff1a\u4f7f\u7528 <code>torch.jit.trace<\/code> \u53ef\u4ee5\u8bb0\u5f55\u6a21\u578b\u7684\u524d\u5411\u4f20\u64ad\u8def\u5f84\uff0c\u4fbf\u4e8e\u540e\u7eed\u90e8\u7f72\u3002<\/li><\/ul>\n\n\n\n<p>\u8fd9\u4e9b\u6a21\u5757\u7ec4\u6210\u4e86 PyTorch \u751f\u6001\u7cfb\u7edf\uff0c\u4e3a\u4ece\u6570\u636e\u52a0\u8f7d\u5230\u6a21\u578b\u8bad\u7ec3\u3001\u90e8\u7f72\u7b49\u5404\u4e2a\u73af\u8282\u63d0\u4f9b\u4e86\u5168\u9762\u7684\u652f\u6301\u3002<br><br><\/p>\n\n\n\n<p>PyTorch \u62e5\u6709\u5f3a\u5927\u7684\u751f\u6001\u7cfb\u7edf\uff0c\u914d\u5408\u5176\u4ed6\u5de5\u5177\u5305\u548c\u6846\u67b6\u53ef\u4ee5\u5b9e\u73b0\u66f4\u4e3a\u5168\u9762\u7684\u6df1\u5ea6\u5b66\u4e60\u5de5\u4f5c\u6d41\uff0c\u4ece\u6570\u636e\u51c6\u5907\u5230\u6a21\u578b\u90e8\u7f72\u7b49\u591a\u65b9\u9762\u90fd\u6709\u5e7f\u6cdb\u5e94\u7528\u3002\u4ee5\u4e0b\u662f\u5e38\u7528\u7684 PyTorch \u751f\u6001\u5de5\u5177\u5305\u548c\u6846\u67b6\uff1a<\/p>\n\n\n\n<h3 class=\"wp-block-heading\">1. \u6570\u636e\u5904\u7406\u548c\u52a0\u8f7d\u5de5\u5177<\/h3>\n\n\n\n<ul class=\"wp-block-list\"><li><strong><code>torchvision<\/code><\/strong>\uff1a\u7528\u4e8e\u8ba1\u7b97\u673a\u89c6\u89c9\u4efb\u52a1\uff0c\u5305\u542b\u5e38\u7528\u6570\u636e\u96c6\uff08\u5982 CIFAR\u3001ImageNet\uff09\u3001\u56fe\u50cf\u5904\u7406\u548c\u6570\u636e\u589e\u5f3a\u5de5\u5177\u3001\u9884\u8bad\u7ec3\u6a21\u578b\u7b49\u3002<\/li><li><strong><code>torchtext<\/code><\/strong>\uff1a\u4e13\u4e3a NLP \u8bbe\u8ba1\uff0c\u5305\u542b\u5e38\u7528\u6587\u672c\u6570\u636e\u96c6\u3001\u8bcd\u6c47\u8868\u548c\u5d4c\u5165\u5c42\u7684\u521b\u5efa\u5de5\u5177\uff0c\u5e76\u652f\u6301\u9884\u8bad\u7ec3\u8bcd\u5d4c\u5165\uff08\u5982 GloVe\u3001FastText\uff09\u3002<\/li><li><strong><code>torchaudio<\/code><\/strong>\uff1a\u4e13\u4e3a\u97f3\u9891\u5904\u7406\u63d0\u4f9b\u7684\u5e93\uff0c\u5305\u542b\u5e38\u7528\u7684\u97f3\u9891\u6570\u636e\u96c6\u3001\u8f6c\u6362\u5de5\u5177\u3001\u7279\u5f81\u63d0\u53d6\u529f\u80fd\uff0c\u5982 Mel \u9891\u8c31\u548c MFCC \u63d0\u53d6\u3002<\/li><li><strong><code>albumentations<\/code><\/strong>\uff1a\u529f\u80fd\u4e30\u5bcc\u7684\u6570\u636e\u589e\u5f3a\u5e93\uff0c\u9002\u7528\u4e8e\u56fe\u50cf\u589e\u5f3a\u548c\u9884\u5904\u7406\uff0c\u53ef\u4e0e PyTorch \u7684 <code>torchvision.transforms<\/code> \u914d\u5408\u4f7f\u7528\u3002<\/li><li><strong><code>pandas<\/code><\/strong> \u548c <strong><code>numpy<\/code><\/strong>\uff1a\u7528\u4e8e\u6570\u636e\u8bfb\u53d6\u3001\u5904\u7406\u548c\u5206\u6790\uff0c\u5c24\u5176\u9002\u5408 CSV\u3001Excel \u6587\u4ef6\u548c\u6570\u503c\u8fd0\u7b97\uff0c\u80fd\u4e0e PyTorch \u5f20\u91cf\u65e0\u7f1d\u8854\u63a5\u3002<\/li><\/ul>\n\n\n\n<h3 class=\"wp-block-heading\">2. \u9ad8\u7ea7\u6a21\u578b\u6784\u5efa\u548c\u8bad\u7ec3\u6846\u67b6<\/h3>\n\n\n\n<ul class=\"wp-block-list\"><li><strong><code>PyTorch Lightning<\/code><\/strong>\uff1a\u7b80\u5316\u4e86 PyTorch \u8bad\u7ec3\u7684\u6d41\u7a0b\uff0c\u5c01\u88c5\u4e86\u8bad\u7ec3\u548c\u9a8c\u8bc1\u903b\u8f91\uff0c\u51cf\u5c11\u4ee3\u7801\u5197\u4f59\u3002\u652f\u6301\u591a GPU \u548c\u5206\u5e03\u5f0f\u8bad\u7ec3\uff0c\u9002\u5408\u5feb\u901f\u5f00\u53d1\u548c\u5b9e\u9a8c\u3002<\/li><li><strong><code>Hugging Face Transformers<\/code><\/strong>\uff1a\u4e3a NLP \u63d0\u4f9b\u4e86\u9884\u8bad\u7ec3\u6a21\u578b\uff08\u5982 BERT\u3001GPT-2\uff09\uff0c\u5e76\u5305\u542b\u591a\u79cd\u6df1\u5ea6\u5b66\u4e60\u4efb\u52a1\u7684\u5de5\u5177\uff0c\u9002\u7528\u4e8e\u60c5\u611f\u5206\u6790\u3001\u7ffb\u8bd1\u3001\u6587\u672c\u751f\u6210\u7b49 NLP \u5e94\u7528\u3002<\/li><li><strong><code>fastai<\/code><\/strong>\uff1a\u57fa\u4e8e PyTorch \u7684\u9ad8\u5c42\u5e93\uff0c\u5c01\u88c5\u4e86\u5e38\u89c1\u6a21\u578b\u548c\u8bad\u7ec3\u6d41\u7a0b\uff0c\u5305\u542b\u4e30\u5bcc\u7684\u56fe\u50cf\u3001\u6587\u672c\u3001\u8868\u683c\u6570\u636e\u5904\u7406\u529f\u80fd\uff0c\u9002\u5408\u5feb\u901f\u5f00\u53d1\u548c\u8c03\u4f18\u3002<\/li><li><strong><code>MONAI<\/code><\/strong>\uff1a\u7528\u4e8e\u533b\u5b66\u5f71\u50cf\u5206\u6790\u7684\u6df1\u5ea6\u5b66\u4e60\u6846\u67b6\uff0c\u652f\u6301\u533b\u7597\u6570\u636e\u9884\u5904\u7406\u3001\u6a21\u578b\u6784\u5efa\u548c\u8bc4\u4f30\uff0c\u7279\u522b\u9002\u5408 MRI\u3001CT \u56fe\u50cf\u5206\u6790\u3002<\/li><\/ul>\n\n\n\n<h3 class=\"wp-block-heading\">3. \u6a21\u578b\u4f18\u5316\u548c\u52a0\u901f<\/h3>\n\n\n\n<ul class=\"wp-block-list\"><li><strong><code>Torch-TensorRT<\/code><\/strong>\uff1a\u7528\u4e8e\u52a0\u901f PyTorch \u6a21\u578b\u5728 NVIDIA GPU \u4e0a\u7684\u63a8\u7406\uff0c\u652f\u6301\u5bf9\u6a21\u578b\u8fdb\u884c\u4f18\u5316\u548c\u8f6c\u6362\uff0c\u4ee5\u63d0\u9ad8\u63a8\u7406\u901f\u5ea6\u3002<\/li><li><strong><code>ONNX (Open Neural Network Exchange)<\/code><\/strong>\uff1a\u53ef\u4ee5\u5c06 PyTorch \u6a21\u578b\u8f6c\u6362\u4e3a\u901a\u7528\u7684 ONNX \u683c\u5f0f\uff0c\u4ee5\u4fbf\u5728\u5176\u4ed6\u5e73\u53f0\uff08\u5982 TensorFlow\u3001Microsoft ML.NET\uff09\u4e0a\u90e8\u7f72\u548c\u52a0\u901f\u63a8\u7406\u3002<\/li><li><strong><code>DeepSpeed<\/code><\/strong>\uff1aMicrosoft \u5f00\u53d1\u7684\u5206\u5e03\u5f0f\u8bad\u7ec3\u5e93\uff0c\u652f\u6301\u6a21\u578b\u5e76\u884c\u548c\u6570\u636e\u5e76\u884c\uff0c\u9002\u5408\u5904\u7406\u8d85\u5927\u89c4\u6a21\u6a21\u578b\uff0c\u5305\u542b\u5185\u5b58\u4f18\u5316\u6280\u672f\u3002<\/li><li><strong><code>NVIDIA Apex<\/code><\/strong>\uff1a\u63d0\u4f9b\u6df7\u5408\u7cbe\u5ea6\u8bad\u7ec3\u5de5\u5177\uff0c\u5229\u7528 16 \u4f4d\u6d6e\u70b9\u8ba1\u7b97\u63d0\u9ad8\u8bad\u7ec3\u6548\u7387\uff0c\u5e76\u51cf\u5c11\u5185\u5b58\u5360\u7528\u3002<\/li><\/ul>\n\n\n\n<h3 class=\"wp-block-heading\">4. \u81ea\u52a8\u5316\u6a21\u578b\u8c03\u53c2\u548c\u641c\u7d22<\/h3>\n\n\n\n<ul class=\"wp-block-list\"><li><strong><code>Optuna<\/code><\/strong>\uff1a\u4e00\u4e2a\u81ea\u52a8\u5316\u8d85\u53c2\u6570\u4f18\u5316\u5e93\uff0c\u9002\u7528\u4e8e\u795e\u7ecf\u7f51\u7edc\u7684\u8d85\u53c2\u6570\u8c03\u4f18\uff0c\u652f\u6301\u5206\u5e03\u5f0f\u641c\u7d22\u3002<\/li><li><strong><code>Ray Tune<\/code><\/strong>\uff1a\u4e00\u4e2a\u5206\u5e03\u5f0f\u8d85\u53c2\u6570\u8c03\u4f18\u5e93\uff0c\u652f\u6301\u591a\u79cd\u641c\u7d22\u7b97\u6cd5\uff0c\u5982\u968f\u673a\u641c\u7d22\u3001\u8d1d\u53f6\u65af\u4f18\u5316\uff0c\u9002\u7528\u4e8e PyTorch \u7b49\u591a\u79cd\u6df1\u5ea6\u5b66\u4e60\u6846\u67b6\u3002<\/li><li><strong><code>Weights &amp; Biases (W&amp;B)<\/code><\/strong> \u548c <strong><code>MLflow<\/code><\/strong>\uff1a\u7528\u4e8e\u5b9e\u9a8c\u7ba1\u7406\u548c\u53ef\u89c6\u5316\uff0c\u53ef\u4ee5\u8bb0\u5f55\u548c\u8ffd\u8e2a\u6a21\u578b\u8d85\u53c2\u6570\u3001\u8bc4\u4f30\u6307\u6807\u3001\u65e5\u5fd7\uff0c\u9002\u5408\u56e2\u961f\u534f\u4f5c\u548c\u5b9e\u9a8c\u590d\u73b0\u3002<\/li><\/ul>\n\n\n\n<h3 class=\"wp-block-heading\">5. \u53ef\u89c6\u5316\u548c\u6a21\u578b\u89e3\u91ca<\/h3>\n\n\n\n<ul class=\"wp-block-list\"><li><strong><code>TensorBoard<\/code><\/strong>\uff1aPyTorch \u539f\u751f\u652f\u6301 TensorBoard\uff0c\u53ef\u4ee5\u53ef\u89c6\u5316\u8bad\u7ec3\u8fc7\u7a0b\u4e2d\u7684\u635f\u5931\u3001\u51c6\u786e\u7387\u548c\u68af\u5ea6\u4fe1\u606f\uff0c\u4fbf\u4e8e\u8c03\u8bd5\u548c\u4f18\u5316\u3002<\/li><li><strong><code>Matplotlib<\/code><\/strong> \u548c <strong><code>Seaborn<\/code><\/strong>\uff1a\u7528\u4e8e\u7ed8\u5236\u7ed3\u679c\u56fe\u8868\uff0c\u8fdb\u884c\u6570\u636e\u548c\u7ed3\u679c\u5206\u6790\u3002<\/li><li><strong><code>Captum<\/code><\/strong>\uff1aPyTorch \u63d0\u4f9b\u7684\u6a21\u578b\u89e3\u91ca\u5de5\u5177\uff0c\u652f\u6301\u591a\u79cd\u89e3\u91ca\u65b9\u6cd5\uff0c\u5982\u96c6\u6210\u68af\u5ea6\u3001\u8f93\u5165\u68af\u5ea6\uff0c\u5e2e\u52a9\u7406\u89e3\u6a21\u578b\u7684\u7279\u5f81\u91cd\u8981\u6027\u3002<\/li><li><strong><code>SHAP<\/code><\/strong> \u548c <strong><code>LIME<\/code><\/strong>\uff1a\u63d0\u4f9b\u901a\u7528\u7684\u6a21\u578b\u89e3\u91ca\u65b9\u6cd5\uff0c\u9002\u5408\u4e8e NLP\u3001\u8ba1\u7b97\u673a\u89c6\u89c9\u7b49\u4efb\u52a1\u7684\u89e3\u91ca\u6027\u5206\u6790\u3002<\/li><\/ul>\n\n\n\n<h3 class=\"wp-block-heading\">6. \u6a21\u578b\u90e8\u7f72\u5de5\u5177<\/h3>\n\n\n\n<ul class=\"wp-block-list\"><li><strong><code>TorchServe<\/code><\/strong>\uff1aPyTorch \u5b98\u65b9\u7684\u6a21\u578b\u90e8\u7f72\u5de5\u5177\uff0c\u7528\u4e8e\u90e8\u7f72\u548c\u7ba1\u7406 PyTorch \u6a21\u578b\uff0c\u53ef\u4ee5\u63d0\u4f9b RESTful API \u670d\u52a1\u63a5\u53e3\u3002<\/li><li><strong><code>ONNX Runtime<\/code><\/strong>\uff1a\u652f\u6301\u5c06 PyTorch \u6a21\u578b\u901a\u8fc7 ONNX \u683c\u5f0f\u8fdb\u884c\u8de8\u5e73\u53f0\u90e8\u7f72\u548c\u52a0\u901f\uff0c\u9002\u7528\u4e8e\u63a8\u7406\u9636\u6bb5\u7684\u4f18\u5316\u3002<\/li><li><strong><code>Flask<\/code> \u548c <code>FastAPI<\/code><\/strong>\uff1aPython \u7684\u8f7b\u91cf\u7ea7 Web \u6846\u67b6\uff0c\u5e38\u7528\u4e8e\u642d\u5efa\u6a21\u578b API \u670d\u52a1\uff0c\u652f\u6301\u5c06 PyTorch \u6a21\u578b\u5feb\u901f\u5c01\u88c5\u4e3a API \u670d\u52a1\u3002<\/li><li><strong><code>Triton Inference Server<\/code><\/strong>\uff1aNVIDIA \u63a8\u51fa\u7684\u6a21\u578b\u63a8\u7406\u670d\u52a1\u5668\uff0c\u652f\u6301\u591a\u79cd\u6df1\u5ea6\u5b66\u4e60\u6846\u67b6\uff08\u5982 PyTorch\u3001TensorFlow\uff09\uff0c\u53ef\u5728 GPU \u4e0a\u9ad8\u6548\u5730\u90e8\u7f72\u6a21\u578b\u3002<\/li><\/ul>\n\n\n\n<h3 class=\"wp-block-heading\">7. \u5de5\u5177\u96c6\u6210\u548c\u534f\u4f5c<\/h3>\n\n\n\n<ul class=\"wp-block-list\"><li><strong><code>Weights &amp; Biases (W&amp;B)<\/code><\/strong>\uff1a\u63d0\u4f9b\u5b9e\u9a8c\u7ba1\u7406\u3001\u53ef\u89c6\u5316\u548c\u6a21\u578b\u7248\u672c\u63a7\u5236\uff0c\u652f\u6301\u56e2\u961f\u534f\u4f5c\u548c\u5b9e\u9a8c\u8bb0\u5f55\uff0c\u9002\u5408\u591a\u9636\u6bb5\u548c\u591a\u7248\u672c\u7684\u5b9e\u9a8c\u7ba1\u7406\u3002<\/li><li><strong><code>Comet.ml<\/code><\/strong>\uff1a\u7528\u4e8e\u5b9e\u9a8c\u7ba1\u7406\u7684\u4e91\u7aef\u5e73\u53f0\uff0c\u652f\u6301\u5b9e\u9a8c\u8ffd\u8e2a\u3001\u8d85\u53c2\u6570\u4f18\u5316\u548c\u591a\u7528\u6237\u534f\u4f5c\u3002<\/li><li><strong><code>DVC (Data Version Control)<\/code><\/strong>\uff1a\u7528\u4e8e\u7ba1\u7406\u6570\u636e\u548c\u6a21\u578b\u7684\u7248\u672c\u63a7\u5236\u5de5\u5177\uff0c\u9002\u5408\u4e8e\u5927\u89c4\u6a21\u6570\u636e\u96c6\u548c\u591a\u7248\u672c\u6a21\u578b\u7684\u9879\u76ee\u3002<\/li><\/ul>\n\n\n\n<p>\u8fd9\u4e9b\u5de5\u5177\u548c\u6846\u67b6\u4e0e PyTorch \u914d\u5408\u4f7f\u7528\uff0c\u4f7f\u6df1\u5ea6\u5b66\u4e60\u5de5\u4f5c\u6d41\u66f4\u9ad8\u6548\u3001\u6a21\u5757\u5316\u4e14\u6613\u4e8e\u7ba1\u7406\uff0c\u4ece\u6570\u636e\u5904\u7406\u3001\u6a21\u578b\u6784\u5efa\u5230\u5b9e\u9a8c\u8ddf\u8e2a\u548c\u6700\u7ec8\u90e8\u7f72\u90fd\u8986\u76d6\u5168\u9762\u3002<\/p>\n","protected":false},"excerpt":{"rendered":"<p>\u5728 PyTorch \u4e2d\uff0c\u4e3b\u8981\u7684\u5e93&#46;&#46;&#46;<\/p>\n","protected":false},"author":1,"featured_media":0,"comment_status":"closed","ping_status":"closed","sticky":false,"template":"","format":"standard","meta":{"footnotes":""},"categories":[47,46,4],"tags":[],"class_list":["post-2438","post","type-post","status-publish","format-standard","hentry","category-cuda","category-pytorch","category-4"],"_links":{"self":[{"href":"https:\/\/sanlangcode.com\/index.php\/wp-json\/wp\/v2\/posts\/2438","targetHints":{"allow":["GET"]}}],"collection":[{"href":"https:\/\/sanlangcode.com\/index.php\/wp-json\/wp\/v2\/posts"}],"about":[{"href":"https:\/\/sanlangcode.com\/index.php\/wp-json\/wp\/v2\/types\/post"}],"author":[{"embeddable":true,"href":"https:\/\/sanlangcode.com\/index.php\/wp-json\/wp\/v2\/users\/1"}],"replies":[{"embeddable":true,"href":"https:\/\/sanlangcode.com\/index.php\/wp-json\/wp\/v2\/comments?post=2438"}],"version-history":[{"count":0,"href":"https:\/\/sanlangcode.com\/index.php\/wp-json\/wp\/v2\/posts\/2438\/revisions"}],"wp:attachment":[{"href":"https:\/\/sanlangcode.com\/index.php\/wp-json\/wp\/v2\/media?parent=2438"}],"wp:term":[{"taxonomy":"category","embeddable":true,"href":"https:\/\/sanlangcode.com\/index.php\/wp-json\/wp\/v2\/categories?post=2438"},{"taxonomy":"post_tag","embeddable":true,"href":"https:\/\/sanlangcode.com\/index.php\/wp-json\/wp\/v2\/tags?post=2438"}],"curies":[{"name":"wp","href":"https:\/\/api.w.org\/{rel}","templated":true}]}}