{"id":21564,"date":"2025-02-20T13:12:24","date_gmt":"2025-02-20T05:12:24","guid":{"rendered":"https:\/\/www.aisharenet.com\/?p=21564"},"modified":"2025-02-20T13:12:40","modified_gmt":"2025-02-20T05:12:40","slug":"colossalai","status":"publish","type":"post","link":"https:\/\/www.kdjingpai.com\/pt\/colossalai\/","title":{"rendered":"ColossalAI\uff1a\u63d0\u4f9b\u9ad8\u6548\u5927\u89c4\u6a21AI\u6a21\u578b\u8bad\u7ec3\u89e3\u51b3\u65b9\u6848"},"content":{"rendered":"<p>ColossalAI\u662f\u7531HPC-AI\u6280\u672f\u516c\u53f8\u5f00\u53d1\u7684\u5f00\u6e90\u5e73\u53f0\uff0c\u65e8\u5728\u4e3a\u5927\u89c4\u6a21AI\u6a21\u578b\u7684\u8bad\u7ec3\u548c\u63a8\u7406\u63d0\u4f9b\u9ad8\u6548\u3001\u4f4e\u6210\u672c\u7684\u89e3\u51b3\u65b9\u6848\u3002\u901a\u8fc7\u652f\u6301\u591a\u79cd\u5e76\u884c\u7b56\u7565\u3001\u5f02\u6784\u5185\u5b58\u7ba1\u7406\u548c\u6df7\u5408\u7cbe\u5ea6\u8bad\u7ec3\uff0cColossalAI\u80fd\u591f\u663e\u8457\u51cf\u5c11\u6a21\u578b\u8bad\u7ec3\u548c\u63a8\u7406\u7684\u65f6\u95f4\u548c\u8d44\u6e90\u6d88\u8017\u3002\u65e0\u8bba\u662f\u5728\u6570\u636e\u5e76\u884c\u3001\u5f20\u91cf\u5e76\u884c\u8fd8\u662f\u6d41\u6c34\u7ebf\u5e76\u884c\u7b49\u65b9\u9762\uff0cColossalAI\u90fd\u63d0\u4f9b\u4e86\u5f3a\u5927\u7684\u5de5\u5177\u548c\u5e93\uff0c\u5e2e\u52a9\u7814\u7a76\u4eba\u5458\u548c\u5f00\u53d1\u8005\u5728\u591aGPU\u96c6\u7fa4\u4e0a\u5b9e\u73b0\u5927\u89c4\u6a21\u6a21\u578b\u7684\u9ad8\u6548\u8bad\u7ec3\u548c\u63a8\u7406\u3002<\/p>\n<p><img loading=\"lazy\" decoding=\"async\" class=\"aligncenter size-full wp-image-21565\" title=\"ColossalAI\uff1a\u63d0\u4f9b\u9ad8\u6548\u5927\u89c4\u6a21AI\u6a21\u578b\u8bad\u7ec3\u89e3\u51b3\u65b9\u6848-1\" src=\"https:\/\/www.kdjingpai.com\/wp-content\/uploads\/2025\/02\/0ee237e927cca3d.png\" alt=\"ColossalAI\uff1a\u63d0\u4f9b\u9ad8\u6548\u5927\u89c4\u6a21AI\u6a21\u578b\u8bad\u7ec3\u89e3\u51b3\u65b9\u6848-1\" width=\"843\" height=\"180\" srcset=\"https:\/\/www.kdjingpai.com\/wp-content\/uploads\/2025\/02\/0ee237e927cca3d.png 843w, https:\/\/www.kdjingpai.com\/wp-content\/uploads\/2025\/02\/0ee237e927cca3d-768x164.png 768w\" sizes=\"auto, (max-width: 843px) 100vw, 843px\" \/><\/p>\n<p>&nbsp;<\/p>\n<h2>\u529f\u80fd\u5217\u8868<\/h2>\n<ul>\n<li>\u6570\u636e\u5e76\u884c\u3001\u5f20\u91cf\u5e76\u884c\u3001\u6d41\u6c34\u7ebf\u5e76\u884c\u7b49\u591a\u79cd\u5e76\u884c\u7b56\u7565\u652f\u6301<\/li>\n<li>\u6df7\u5408\u7cbe\u5ea6\u8bad\u7ec3\u548c\u96f6\u5197\u4f59\u4f18\u5316\u5668\uff08ZeRO\uff09<\/li>\n<li>\u5f02\u6784\u5185\u5b58\u7ba1\u7406\uff0c\u652f\u6301\u5927\u6a21\u578b\u7684\u9ad8\u6548\u8bad\u7ec3<\/li>\n<li>\u652f\u6301\u591a\u79cd\u9886\u57df\u7279\u5b9a\u6a21\u578b\uff0c\u5982Open-Sora\u3001Colossal-LLaMA\u7b49<\/li>\n<li>\u63d0\u4f9b\u5206\u5e03\u5f0f\u8bad\u7ec3\u548c\u63a8\u7406\u7684\u7528\u6237\u53cb\u597d\u5de5\u5177<\/li>\n<li>\u96c6\u6210\u9ad8\u6027\u80fd\u5185\u6838\u3001KV\u7f13\u5b58\u3001\u5206\u9875\u6ce8\u610f\u529b\u548c\u8fde\u7eed\u6279\u5904\u7406\u7b49\u6280\u672f<\/li>\n<li>\u901a\u8fc7\u914d\u7f6e\u6587\u4ef6\u5b9e\u73b0\u5e76\u884c\u8bad\u7ec3\u7684\u7b80\u4fbf\u914d\u7f6e<\/li>\n<li>\u63d0\u4f9b\u4e30\u5bcc\u7684\u793a\u4f8b\u548c\u6587\u6863\uff0c\u52a9\u529b\u5feb\u901f\u4e0a\u624b<\/li>\n<li>\u63d0\u4f9bDocker\u955c\u50cf\u548c\u4ece\u6e90\u7801\u6784\u5efa\u7684\u591a\u79cd\u5b89\u88c5\u65b9\u5f0f<\/li>\n<\/ul>\n<p>&nbsp;<\/p>\n<h2>\u4f7f\u7528\u5e2e\u52a9<\/h2>\n<h3>\u5b89\u88c5\u6307\u5357<\/h3>\n<h4>\u4ecePyPI\u5b89\u88c5<\/h4>\n<p>\u60a8\u53ef\u4ee5\u901a\u8fc7\u4ee5\u4e0b\u547d\u4ee4\u8f7b\u677e\u5b89\u88c5Colossal-AI\uff1a<\/p>\n<pre><code>pip install colossalai\r\n<\/code><\/pre>\n<p>\u9ed8\u8ba4\u60c5\u51b5\u4e0b\uff0c\u4e0d\u4f1a\u5728\u5b89\u88c5\u8fc7\u7a0b\u4e2d\u6784\u5efaPyTorch\u6269\u5c55\u3002\u5982\u679c\u9700\u8981\u6784\u5efaPyTorch\u6269\u5c55\uff0c\u53ef\u4ee5\u8bbe\u7f6e<code>BUILD_EXT=1<\/code>\uff1a<\/p>\n<pre><code>BUILD_EXT=1 pip install colossalai\r\n<\/code><\/pre>\n<p>\u6b64\u5916\uff0c\u6211\u4eec\u6bcf\u5468\u90fd\u4f1a\u53d1\u5e03nightly\u7248\u672c\uff0c\u5141\u8bb8\u60a8\u8bbf\u95ee\u6700\u65b0\u7684\u672a\u53d1\u5e03\u529f\u80fd\u548cbug\u4fee\u590d\u3002\u5b89\u88c5\u65b9\u5f0f\u5982\u4e0b\uff1a<\/p>\n<pre><code>pip install colossalai-nightly\r\n<\/code><\/pre>\n<h4>\u4ece\u6e90\u7801\u5b89\u88c5<\/h4>\n<pre><code>git clone https:\/\/github.com\/hpcaitech\/ColossalAI.git\r\ncd ColossalAI\r\npip install .\r\n<\/code><\/pre>\n<p>\u9ed8\u8ba4\u60c5\u51b5\u4e0b\u4e0d\u4f1a\u7f16\u8bd1CUDA\/C++\u5185\u6838\u3002ColossalAI\u5c06\u5728\u8fd0\u884c\u65f6\u6784\u5efa\u5b83\u4eec\u3002\u5982\u679c\u9700\u8981\u542f\u7528CUDA\u5185\u6838\u878d\u5408\uff1a<\/p>\n<pre><code>BUILD_EXT=1 pip install .\r\n<\/code><\/pre>\n<p>\u5bf9\u4e8eCUDA 10.2\u7528\u6237\uff0c\u53ef\u4ee5\u624b\u52a8\u4e0b\u8f7dcub\u5e93\u5e76\u5c06\u5176\u590d\u5236\u5230\u76f8\u5e94\u76ee\u5f55\u540e\u518d\u8fdb\u884c\u5b89\u88c5\u3002<\/p>\n<h3>\u4f7f\u7528Docker<\/h3>\n<h4>\u4eceDockerHub\u62c9\u53d6\u955c\u50cf<\/h4>\n<p>\u60a8\u53ef\u4ee5\u76f4\u63a5\u4ece<a href=\"https:\/\/hub.docker.com\/r\/hpcaitech\/colossalai\">DockerHub\u9875\u9762<\/a>\u62c9\u53d6Docker\u955c\u50cf\u3002<\/p>\n<h4>\u81ea\u884c\u6784\u5efa\u955c\u50cf<\/h4>\n<pre><code>cd ColossalAI\r\ndocker build -t colossalai .\/docker\r\n<\/code><\/pre>\n<p>\u542f\u52a8\u4ea4\u4e92\u6a21\u5f0f\u7684\u5bb9\u5668\uff1a<\/p>\n<pre><code>docker run -ti --gpus all --rm --ipc=host colossalai bash\r\n<\/code><\/pre>\n<h3>\u529f\u80fd\u64cd\u4f5c\u6d41\u7a0b<\/h3>\n<h4>\u6570\u636e\u5e76\u884c<\/h4>\n<p>\u6570\u636e\u5e76\u884c\u662f\u5c06\u6570\u636e\u96c6\u5212\u5206\u6210\u591a\u4e2a\u5b50\u96c6\uff0c\u5e76\u5728\u591a\u4e2aGPU\u4e0a\u5e76\u884c\u8bad\u7ec3\u6a21\u578b\u3002ColossalAI\u901a\u8fc7\u7b80\u5316\u7684\u6570\u636e\u5e76\u884c\u914d\u7f6e\u6587\u4ef6\uff0c\u4f7f\u5f97\u7528\u6237\u80fd\u591f\u8f7b\u677e\u5b9e\u73b0\u6570\u636e\u5e76\u884c\u8bad\u7ec3\uff1a<\/p>\n<pre><code>from colossalai.nn.parallel import DataParallel\r\nmodel = DataParallel(model)\r\n<\/code><\/pre>\n<h4>\u5f20\u91cf\u5e76\u884c<\/h4>\n<p>\u5f20\u91cf\u5e76\u884c\u662f\u5c06\u6a21\u578b\u7684\u53c2\u6570\u5f20\u91cf\u5212\u5206\u6210\u591a\u4e2a\u5b50\u5f20\u91cf\uff0c\u5e76\u5728\u591a\u4e2aGPU\u4e0a\u5e76\u884c\u8ba1\u7b97\u3002ColossalAI\u63d0\u4f9b\u4e861D\u30012D\u30012.5D\u548c3D\u5f20\u91cf\u5e76\u884c\u7684\u5b9e\u73b0\uff1a<\/p>\n<pre><code>from colossalai.nn.parallel import TensorParallel\r\nmodel = TensorParallel(model, parallel_mode='1D')\r\n<\/code><\/pre>\n<h4>\u6d41\u6c34\u7ebf\u5e76\u884c<\/h4>\n<p>\u6d41\u6c34\u7ebf\u5e76\u884c\u662f\u5c06\u6a21\u578b\u5206\u6210\u591a\u4e2a\u9636\u6bb5\uff0c\u6bcf\u4e2a\u9636\u6bb5\u7531\u4e00\u4e2a\u6216\u591a\u4e2aGPU\u6267\u884c\u3002ColossalAI\u63d0\u4f9b\u4e86\u7b80\u4fbf\u7684\u6d41\u6c34\u7ebf\u5e76\u884c\u914d\u7f6e\uff1a<\/p>\n<pre><code>from colossalai.pipeline.parallel import PipelineParallel\r\nmodel = PipelineParallel(model, num_stages=4)\r\n<\/code><\/pre>\n<h4>\u6df7\u5408\u7cbe\u5ea6\u8bad\u7ec3<\/h4>\n<p>\u6df7\u5408\u7cbe\u5ea6\u8bad\u7ec3\u901a\u8fc7\u5728\u8bad\u7ec3\u8fc7\u7a0b\u4e2d\u4f7f\u752816\u4f4d\u6d6e\u70b9\u6570\uff08FP16\uff09\u548c32\u4f4d\u6d6e\u70b9\u6570\uff08FP32\uff09\u76f8\u7ed3\u5408\uff0c\u80fd\u591f\u663e\u8457\u51cf\u5c11\u663e\u5b58\u4f7f\u7528\u5e76\u52a0\u5feb\u8bad\u7ec3\u901f\u5ea6\uff1a<\/p>\n<pre><code>from colossalai.amp import convert_to_amp\r\nmodel, optimizer, criterion = convert_to_amp(model, optimizer, criterion)\r\n<\/code><\/pre>\n<h4>\u96f6\u5197\u4f59\u4f18\u5316\u5668\uff08ZeRO\uff09<\/h4>\n<p>ZeRO\u4f18\u5316\u5668\u901a\u8fc7\u5c06\u4f18\u5316\u5668\u72b6\u6001\u3001\u68af\u5ea6\u548c\u53c2\u6570\u5206\u5e03\u5230\u591a\u4e2aGPU\u4e0a\uff0c\u663e\u8457\u51cf\u5c11\u4e86\u663e\u5b58\u5360\u7528\uff1a<\/p>\n<pre><code>from colossalai.zero import ZeroOptimizer\r\noptimizer = ZeroOptimizer(optimizer, model)\r\n<\/code><\/pre>\n<h3>\u771f\u5b9e\u4e16\u754c\u5e94\u7528<\/h3>\n<h4>Open-Sora<\/h4>\n<p>Open-Sora\u662fColossalAI\u4e3a\u89c6\u9891\u751f\u6210\u6a21\u578b\u63d0\u4f9b\u7684\u4e00\u5957\u5b8c\u6574\u89e3\u51b3\u65b9\u6848\uff0c\u5305\u62ec\u6a21\u578b\u53c2\u6570\u3001\u8bad\u7ec3\u7ec6\u8282\u548c\u4e00\u952e\u751f\u621016\u79d2720p\u9ad8\u6e05\u89c6\u9891\u7684\u529f\u80fd\uff1a<\/p>\n<pre><code># \u8bad\u7ec3\r\npython train.py\r\n# \u63a8\u7406\r\npython infer.py\r\n<\/code><\/pre>\n<p>\u8be6\u7ec6\u4fe1\u606f\u8bf7\u53c2\u89c1<a href=\"https:\/\/github.com\/hpcaitech\/Open-Sora\">Open-Sora<\/a>\u3002<\/p>\n<h4>Colossal-LLaMA<\/h4>\n<p>Colossal-LLaMA\u63d0\u4f9b\u4e86\u4e00\u4e2a\u5f00\u6e90\u7684\u9886\u57df\u7279\u5b9a\u5927\u8bed\u8a00\u6a21\u578b\uff08LLM\uff09\u89e3\u51b3\u65b9\u6848\uff0c\u901a\u8fc7\u5c11\u91cf\u8bad\u7ec3\u8d44\u91d1\u5373\u53ef\u83b7\u5f97\u4e0e\u4e3b\u6d41\u5927\u6a21\u578b\u76f8\u5f53\u7684\u6548\u679c\uff1a<\/p>\n<pre><code># \u8bad\u7ec3\r\npython train_llama.py\r\n# \u63a8\u7406\r\npython infer_llama.py\r\n<\/code><\/pre>\n<p>\u8be6\u7ec6\u4fe1\u606f\u8bf7\u53c2\u89c1<a href=\"https:\/\/github.com\/hpcaitech\/ColossalAI\/tree\/main\/applications\/Colossal-LLaMA-2\">Colossal-LLaMA<\/a>\u3002<\/p>\n","protected":false},"excerpt":{"rendered":"<p>ColossalAI\u662f\u7531HPC-AI\u6280\u672f\u516c\u53f8\u5f00\u53d1\u7684\u5f00\u6e90\u5e73\u53f0\uff0c\u65e8\u5728\u4e3a\u5927\u89c4\u6a21AI\u6a21\u578b\u7684\u8bad\u7ec3\u548c\u63a8\u7406\u63d0\u4f9b\u9ad8\u6548\u3001\u4f4e\u6210\u672c\u7684\u89e3\u51b3\u65b9\u6848\u3002\u901a\u8fc7\u652f\u6301\u591a\u79cd\u5e76\u884c\u7b56\u7565\u3001\u5f02\u6784\u5185\u5b58\u7ba1\u7406\u548c\u6df7\u5408\u7cbe\u5ea6\u8bad\u7ec3\uff0cColossalAI\u80fd\u591f\u663e\u8457\u51cf\u5c11\u6a21\u578b\u8bad\u7ec3\u548c\u63a8\u7406\u7684\u65f6\u95f4\u548c\u8d44\u6e90\u6d88\u8017\u3002\u65e0\u8bba\u662f&#8230;<\/p>\n","protected":false},"author":1,"featured_media":61869,"comment_status":"open","ping_status":"closed","sticky":false,"template":"","format":"standard","meta":{"footnotes":""},"categories":[20],"tags":[230,365],"class_list":["post-21564","post","type-post","status-publish","format-standard","has-post-thumbnail","hentry","category-tool","tag-aikaiyuanxiangmu","tag-damoxingweidiao"],"_links":{"self":[{"href":"https:\/\/www.kdjingpai.com\/pt\/wp-json\/wp\/v2\/posts\/21564","targetHints":{"allow":["GET"]}}],"collection":[{"href":"https:\/\/www.kdjingpai.com\/pt\/wp-json\/wp\/v2\/posts"}],"about":[{"href":"https:\/\/www.kdjingpai.com\/pt\/wp-json\/wp\/v2\/types\/post"}],"author":[{"embeddable":true,"href":"https:\/\/www.kdjingpai.com\/pt\/wp-json\/wp\/v2\/users\/1"}],"replies":[{"embeddable":true,"href":"https:\/\/www.kdjingpai.com\/pt\/wp-json\/wp\/v2\/comments?post=21564"}],"version-history":[{"count":0,"href":"https:\/\/www.kdjingpai.com\/pt\/wp-json\/wp\/v2\/posts\/21564\/revisions"}],"wp:featuredmedia":[{"embeddable":true,"href":"https:\/\/www.kdjingpai.com\/pt\/wp-json\/wp\/v2\/media\/61869"}],"wp:attachment":[{"href":"https:\/\/www.kdjingpai.com\/pt\/wp-json\/wp\/v2\/media?parent=21564"}],"wp:term":[{"taxonomy":"category","embeddable":true,"href":"https:\/\/www.kdjingpai.com\/pt\/wp-json\/wp\/v2\/categories?post=21564"},{"taxonomy":"post_tag","embeddable":true,"href":"https:\/\/www.kdjingpai.com\/pt\/wp-json\/wp\/v2\/tags?post=21564"}],"curies":[{"name":"wp","href":"https:\/\/api.w.org\/{rel}","templated":true}]}}