{"id":35955,"date":"2025-01-03T14:00:10","date_gmt":"2025-01-03T06:00:10","guid":{"rendered":"https:\/\/17aitech.com\/?p=35955"},"modified":"2025-01-03T14:00:10","modified_gmt":"2025-01-03T06:00:10","slug":"%e5%be%ae%e8%b0%83%e5%a4%a7%e6%a8%a1%e5%9e%8b%ef%bc%8camd-mi300x%e5%b0%b1%e5%a4%9f%e4%ba%86%ef%bc%81%e8%b7%9f%e7%9d%80%e8%bf%99%e7%af%87%e5%8d%9a%e5%ae%a2%e5%be%ae%e8%b0%83llama-3-1-405b%ef%bc%8c","status":"publish","type":"post","link":"https:\/\/17aitech.com\/?p=35955","title":{"rendered":"\u5fae\u8c03\u5927\u6a21\u578b\uff0cAMD MI300X\u5c31\u591f\u4e86\uff01\u8ddf\u7740\u8fd9\u7bc7\u535a\u5ba2\u5fae\u8c03Llama 3.1 405B\uff0c\u6548\u679c\u5ab2\u7f8eH100"},"content":{"rendered":"<p>\u6587\u7ae0\u6765\u6e90\u4e8e\u4e92\u8054\u7f51:<a href=\"https:\/\/www.jiqizhixin.com\/articles\/2024-10-08-4\" target=\"_blank\">\u5fae\u8c03\u5927\u6a21\u578b\uff0cAMD MI300X\u5c31\u591f\u4e86\uff01\u8ddf\u7740\u8fd9\u7bc7\u535a\u5ba2\u5fae\u8c03Llama 3.1 405B\uff0c\u6548\u679c\u5ab2\u7f8eH100<\/a><\/p>\n<section>\u968f\u7740 AI \u6a21\u578b\u7684\u53c2\u6570\u91cf\u8d8a\u6765\u8d8a\u5927\uff0c\u5bf9\u7b97\u529b\u7684\u9700\u6c42\u4e5f\u6c34\u6da8\u8239\u9ad8\u3002<\/section>\n<section><\/section>\n<section>\u6bd4\u5982\u6700\u8fd1\uff0cLlama-3.1 \u767b\u4e0a\u4e86\u6700\u5f3a\u5f00\u6e90\u5927\u6a21\u578b\u7684\u5b9d\u5ea7\uff0c\u4f46\u8d85\u5927\u676f 405B \u7248\u672c\u7684\u5185\u5b58\u5c31\u9ad8\u8fbe 900 \u591a GB\uff0c\u8fd9\u5bf9\u7b97\u529b\u6784\u6210\u4e86\u66f4\u52a0\u82db\u523b\u7684\u6311\u6218\u3002<\/section>\n<section><\/section>\n<section>\u5982\u4f55\u964d\u4f4e\u7b97\u529b\u7684\u4f7f\u7528\u6210\u672c\u548c\u4f7f\u7528\u95e8\u69db\uff0c\u5df2\u7ecf\u6210\u4e3a\u8bb8\u591a\u516c\u53f8\u5bfb\u6c42\u7a81\u7834\u7684\u5173\u952e\u3002Felafax \u5c31\u662f\u5176\u4e2d\u7684\u4e00\u5bb6\u521b\u4e1a\u516c\u53f8\uff0c\u81f4\u529b\u4e8e\u7b80\u5316 AI \u8bad\u7ec3\u96c6\u7fa4\u7684\u642d\u5efa\u6d41\u7a0b\u3002<\/section>\n<section><a href=\"https:\/\/17aitech.com\/wp-content\/uploads\/2024\/10\/frc-40479d1ba65cdc53f5e58dc64a913fec.png\" data-fancybox=\"images\" data-fancybox=\"gallery\"><img decoding=\"async\" src=\"https:\/\/17aitech.com\/wp-content\/uploads\/2024\/10\/frc-40479d1ba65cdc53f5e58dc64a913fec.png\"><\/a><\/section>\n<section><em><sup>\u00a0 \u00a0 \u00a0 \u00a0 \u00a0 \u00a0 \u00a0 \u00a0 \u00a0 \u00a0 \u00a0 \u00a0Nikhil Sonti \u548c Nikhin Sonti \u521b\u7acb\u4e86 Felafax\uff0c\u4ed6\u4eec\u7684\u53e3\u53f7\u662f\u5728\u6784\u5efa\u5f00\u6e90 AI \u5e73\u53f0\uff0c\u4e3a\u4e0b\u4e00\u4ee3 AI \u786c\u4ef6\u670d\u52a1\uff0c\u5c06\u673a\u5668\u5b66\u4e60\u7684\u8bad\u7ec3\u6210\u672c\u964d\u4f4e 30%\u3002<\/sup><\/em><\/section>\n<section><\/section>\n<section>\u4e0e\u82f1\u4f1f\u8fbe\u76f8\u6bd4\uff0cAMD \u7684 GPU\uff0c\u5c24\u5176\u662f MI300X \u7cfb\u5217\uff0c\u63d0\u4f9b\u4e86\u66f4\u9ad8\u7684\u6027\u4ef7\u6bd4\uff0c\u6309\u6bcf\u7f8e\u5143\u8ba1\u7b97\uff0c\u5176\u6027\u80fd\u8868\u73b0\u66f4\u4e3a\u51fa\u8272\u3002<\/section>\n<section><\/section>\n<section>\u6700\u8fd1\uff0cFelafax \u7684\u8054\u5408\u521b\u59cb\u4eba Nikhil Sonti \u53d1\u5e03\u4e86\u4e00\u7bc7\u535a\u5ba2\uff0c\u8be6\u7ec6\u5206\u4eab\u4e86\u5982\u4f55\u901a\u8fc7 8 \u5f20 AMD MI300X GPU \u548c JAX \u5fae\u8c03 LLaMA 3.1 405B \u6a21\u578b\u7684\u65b9\u6cd5\uff0c\u6240\u6709\u4ee3\u7801\u73b0\u5df2\u5f00\u6e90\u3002<\/section>\n<p><a href=\"https:\/\/17aitech.com\/wp-content\/uploads\/2024\/10\/frc-4cbd0a672cfc4b174b91cbe72c672691.png\" data-fancybox=\"images\" data-fancybox=\"gallery\"><img decoding=\"async\" src=\"https:\/\/17aitech.com\/wp-content\/uploads\/2024\/10\/frc-4cbd0a672cfc4b174b91cbe72c672691.png\"><\/a><\/p>\n<section>Github \u94fe\u63a5\uff1ahttps:\/\/github.com\/felafax\/felafax<\/section>\n<section><\/section>\n<section>\u673a\u5668\u4e4b\u5fc3\u5bf9\u535a\u5ba2\u5185\u5bb9\u8fdb\u884c\u4e86\u4e0d\u6539\u53d8\u539f\u610f\u7684\u7f16\u8bd1\u3001\u6574\u7406\uff0c\u4ee5\u4e0b\u662f\u535a\u5ba2\u5185\u5bb9\uff1a<\/section>\n<section><\/section>\n<section><strong>JAX \u5c24\u5176\u9002\u5408\u975e\u82f1\u4f1f\u8fbe\u786c\u4ef6<\/strong><\/section>\n<section><\/section>\n<section>JAX \u662f\u4e00\u4e2a\u5f3a\u5927\u7684\u673a\u5668\u5b66\u4e60\u5e93\uff0c\u7ed3\u5408\u4e86\u7c7b\u4f3c NumPy \u7684 API\u3001\u81ea\u52a8\u5fae\u5206\u529f\u80fd\u4ee5\u53ca Google \u7684 XLA \u7f16\u8bd1\u5668\u3002\u5b83\u5728\u6a21\u578b\u5e76\u884c\u5316\u65b9\u9762\u63d0\u4f9b\u4e86\u4f18\u79c0\u7684 API\uff0c\u56e0\u6b64\u975e\u5e38\u9002\u5408\u50cf LLaMA 3.1 405B \u8fd9\u6837\u7684\u8d85\u5927\u6a21\u578b\u8bad\u7ec3\u3002<\/section>\n<section><\/section>\n<section>\u5728\u4f7f\u7528 AMD \u786c\u4ef6\u65f6\uff0cJAX \u6709\u51e0\u4e2a\u660e\u663e\u7684\u4f18\u52bf\uff1a<\/section>\n<section><\/section>\n<ul>\n<li>\n<section><strong>\u591a\u786c\u4ef6\u5e76\u884c\u652f\u6301<\/strong>\uff1aJAX \u91c7\u7528 XLA\uff08\u52a0\u901f\u7ebf\u6027\u4ee3\u6570\uff09\u7f16\u8bd1\u5668\uff0c\u5c06\u8ba1\u7b97\u7f16\u8bd1\u4e3a\u786c\u4ef6\u65e0\u5173\u7684\u4e2d\u95f4\u8868\u793a\uff08HLO\uff09\uff0c\u8fd9\u610f\u5473\u7740\u540c\u6837\u7684 JAX \u4ee3\u7801\u65e0\u9700\u4fee\u6539\u4fbf\u53ef\u9ad8\u6548\u8fd0\u884c\u5728\u4e0d\u540c\u786c\u4ef6\u540e\u7aef\uff0c\u5305\u62ec AMD GPU\u3002<\/section>\n<\/li>\n<li>\n<section><strong>\u72ec\u7acb\u4e8e\u5e95\u5c42\u786c\u4ef6<\/strong>\uff1aXLA \u7f16\u8bd1\u5668\u7684\u4f18\u5316\u7b56\u7565\u662f\u901a\u7528\u7684\uff0c\u4e0d\u9488\u5bf9\u67d0\u4e2a\u7279\u5b9a\u7684\u786c\u4ef6\u5e73\u53f0\u3002\u8fd9\u4f7f\u5f97\u4efb\u4f55\u652f\u6301 XLA \u7684\u786c\u4ef6\u8bbe\u5907\uff08\u5982 CPU\u3001GPU\u3001TPU\uff09\u90fd\u80fd\u53d7\u76ca\u4e8e\u8fd9\u4e9b\u4f18\u5316\uff0c\u83b7\u5f97\u66f4\u597d\u7684\u6027\u80fd\u8868\u73b0\u3002<\/section>\n<\/li>\n<li>\n<section><strong>\u6781\u9ad8\u7684<\/strong><strong>\u9002\u5e94\u6027<\/strong>\uff1a\u4ece NVIDIA \u8f6c\u79fb\u5230 AMD\uff08\u6216\u5176\u4ed6\u786c\u4ef6\uff09\u65f6\uff0cJAX \u53ea\u9700\u505a\u6781\u5c11\u7684\u4ee3\u7801\u6539\u52a8\u3002\u800c\u76f8\u8f83\u4e4b\u4e0b\uff0cPyTorch \u4e0e\u82f1\u4f1f\u8fbe\u7684 CUDA \u751f\u6001\u7cfb\u7edf\u7d27\u5bc6\u8026\u5408\uff0c\u8fc1\u79fb\u8fc7\u7a0b\u76f8\u5bf9\u590d\u6742\u3002<\/section>\n<\/li>\n<\/ul>\n<section><\/section>\n<section>\u56e0\u6b64\uff0cJAX \u6210\u4e3a\u4e86\u6211\u4eec\u5728\u975e\u82f1\u4f1f\u8fbe\u786c\u4ef6\u4e0a\u7684\u6700\u4f73\u9009\u62e9\u3002<\/section>\n<section><\/section>\n<section><strong>\u62c9\u53d6 Docker \u955c\u50cf\uff1a<\/strong><\/section>\n<section>\n<pre data-lang=\"nginx\"><code>docker pull rocm\/jax:latest<\/code><\/pre>\n<\/section>\n<section><strong>\u542f\u52a8 Docker \u5bb9\u5668\uff1a<\/strong><\/section>\n<section>\n<pre data-lang=\"properties\"><code># Pull the Docker Image:<\/code>\r\n<code>docker pull rocm\/jax:latest <\/code>\r\n<code>\r\n<\/code><code># Start the Docker Container:<\/code>\r\n<code>docker run -it -w \/workspace --device=\/dev\/kfd --device=\/dev\/dri --group-add video <\/code>\r\n<code>--cap-add=SYS_PTRACE --security-opt seccomp=unconfined --shm-size 16G rocm\/jax:latest<\/code>\r\n<code>\r\n<\/code><code># Verify the Installation: <\/code>\r\n<code>python3 -c 'import jax; print(jax.devices())'<\/code><\/pre>\n<\/section>\n<section><strong>\u9a8c\u8bc1\u5b89\u88c5<\/strong><\/section>\n<section>\n<pre data-lang=\"swift\"><code>python3 -c 'import jax; print (jax.devices ())'<\/code><\/pre>\n<\/section>\n<section>\u8bad\u7ec3\u4f7f\u7528\u4e86\u4e00\u4e2a\u914d\u5907\u4e86 8 \u5f20 AMD MI300x GPU \u7684 AMD \u8282\u70b9\u3002\u6bcf\u5f20 MI300x \u62e5\u6709 192GB \u7684 HBM3 \u5185\u5b58\uff0c\u6027\u80fd\u8868\u73b0\u4e0e\u6700\u65b0\u7684\u82f1\u4f1f\u8fbe H100 GPU \u76f8\u6bd4\u975e\u5e38\u51fa\u8272\u3002<\/section>\n<section><a href=\"https:\/\/17aitech.com\/wp-content\/uploads\/2024\/10\/frc-c53d4a00a5ea8e662b790c9443db671f.png\" data-fancybox=\"images\" data-fancybox=\"gallery\"><img decoding=\"async\" src=\"https:\/\/17aitech.com\/wp-content\/uploads\/2024\/10\/frc-c53d4a00a5ea8e662b790c9443db671f.png\"><\/a><\/section>\n<section><em><sup>\u00a0 \u00a0 \u00a0 \u00a0 \u00a0 \u00a0 \u00a0 \u00a0 \u00a0 \u00a0 \u00a0 \u00a0 \u00a0 \u00a0 \u00a0 \u4e0e\u82f1\u4f1f\u8fbe H100 \u7684\u6bd4\u8f83\uff0c\u6765\u6e90\uff1aTensorWave<\/sup><\/em><\/section>\n<section><\/section>\n<section><strong>\u8bad\u7ec3 LLaMA 405B\uff1a\u6027\u80fd\u4e0e\u53ef\u6269\u5c55\u6027<\/strong><\/section>\n<section><\/section>\n<section>\u4f7f\u7528 JAX\uff0c\u53ef\u4ee5\u6210\u529f\u5730\u5728 AMD GPU \u4e0a\u8bad\u7ec3 LLaMA 405B \u6a21\u578b\u3002\u6211\u4eec\u4f7f\u7528 LoRA \u5fae\u8c03\uff0c\u5c06\u6240\u6709\u6a21\u578b\u6743\u91cd\u548c LoRA \u53c2\u6570\u90fd\u8bbe\u4e3a bfloat16\uff0cLoRA rank \u8bbe\u4e3a 8\uff0cLoRA alpha \u8bbe\u4e3a 16\uff1a<\/section>\n<section><\/section>\n<ul>\n<li>\n<section><strong>\u6a21\u578b\u5927\u5c0f<\/strong>\uff1aLLaMA \u6a21\u578b\u7684\u6743\u91cd\u5360\u7528\u4e86\u7ea6 800GB \u7684\u663e\u5b58\u3002<\/section>\n<\/li>\n<li>\n<section><strong>LoRA \u6743\u91cd + \u4f18\u5316\u5668\u72b6\u6001<\/strong>\uff1a\u5927\u7ea6\u5360\u7528\u4e86 400GB \u7684\u663e\u5b58\u3002<\/section>\n<\/li>\n<li>\n<section><strong>\u663e\u5b58\u603b\u4f7f\u7528\u91cf<\/strong>\uff1a\u5360\u603b\u663e\u5b58\u7684 77%\uff0c\u7ea6 1200GB\u3002<\/section>\n<\/li>\n<li>\n<section><strong>\u9650\u5236<\/strong>\uff1a\u7531\u4e8e 405B \u6a21\u578b\u7684\u89c4\u6a21\u8fc7\u5927\uff0cbatch \u5927\u5c0f\u548c\u5e8f\u5217\u957f\u5ea6\u7684\u7a7a\u95f4\u6709\u9650\uff0c\u4f7f\u7528\u7684 batch size \u4e3a 16\uff0c\u5e8f\u5217\u957f\u5ea6\u4e3a 64\u3002<\/section>\n<\/li>\n<li>\n<section><strong>JIT \u7f16\u8bd1<\/strong>\uff1a\u7531\u4e8e\u7a7a\u95f4\u9650\u5236\uff0c\u65e0\u6cd5\u8fd0\u884c JIT \u7f16\u8bd1\u7248\u672c\uff1b\u5b83\u53ef\u80fd\u9700\u8981\u6bd4\u6025\u5207\u6a21\u5f0f\u7a0d\u591a\u7684\u7a7a\u95f4\u3002<\/section>\n<\/li>\n<li>\n<section><strong>\u8bad\u7ec3\u901f\u5ea6<\/strong>\uff1a\u4f7f\u7528 JAX \u6025\u5207\u6a21\u5f0f\uff0c\u7ea6\u4e3a 35 tokens \/ \u79d2\u3002<\/section>\n<\/li>\n<li>\n<section><strong>\u5185\u5b58\u6548\u7387<\/strong>\uff1a\u7a33\u5b9a\u5728\u7ea6 70% \u5de6\u53f3\u3002<\/section>\n<\/li>\n<li>\n<section><strong>\u6269\u5c55\u6027<\/strong>\uff1a\u5728 8 \u5f20 GPU \u4e0a\uff0c\u4f7f\u7528 JAX \u7684\u6269\u5c55\u6027\u63a5\u8fd1\u7ebf\u6027\u3002<\/section>\n<\/li>\n<\/ul>\n<section><\/section>\n<section>\u7531\u4e8e\u786c\u4ef6\u548c\u663e\u5b58\u7684\u9650\u5236\uff0c\u6211\u4eec\u65e0\u6cd5\u8fd0\u884c JIT \u7f16\u8bd1\u7248\u672c\u7684 405B \u6a21\u578b\uff0c\u6574\u4e2a\u8bad\u7ec3\u8fc7\u7a0b\u662f\u5728 JAX \u7684\u6025\u5207\u6a21\u5f0f\u4e0b\u6267\u884c\u7684\uff0c\u56e0\u6b64\u8fd8\u6709\u5f88\u5927\u7684\u8fdb\u6b65\u7a7a\u95f4\u3002\u00a0<\/section>\n<section><\/section>\n<section>\u4e0b\u56fe\u4e2d\u663e\u793a\u4e86\u5728\u4e00\u6b21\u5fae\u8c03\u8bad\u7ec3\u6b65\u9aa4\u4e2d\uff0c8 \u5f20 GPU \u7684\u663e\u5b58\u5229\u7528\u7387\u548c rocm-smi \u8f93\u51fa\uff1a<\/section>\n<section><\/section>\n<section>GPU \u5229\u7528\u7387\uff1a<\/section>\n<section><a href=\"https:\/\/17aitech.com\/wp-content\/uploads\/2024\/10\/frc-ce14e828b381620975dff8e017ad7162.png\" data-fancybox=\"images\" data-fancybox=\"gallery\"><img decoding=\"async\" src=\"https:\/\/17aitech.com\/wp-content\/uploads\/2024\/10\/frc-ce14e828b381620975dff8e017ad7162.png\"><\/a><\/section>\n<section>\u663e\u5b58\u5229\u7528\u7387\uff1a<\/section>\n<section><a href=\"https:\/\/17aitech.com\/wp-content\/uploads\/2024\/10\/frc-7c76a5ba04a78dd34273d7b410ab12a4.png\" data-fancybox=\"images\" data-fancybox=\"gallery\"><img decoding=\"async\" src=\"https:\/\/17aitech.com\/wp-content\/uploads\/2024\/10\/frc-7c76a5ba04a78dd34273d7b410ab12a4.png\"><\/a><\/section>\n<section>rocm-smi \u8f93\u51fa\uff1a<\/section>\n<p><a href=\"https:\/\/17aitech.com\/wp-content\/uploads\/2024\/10\/frc-4eea268e2aa42a494afea7ea5e0ffcac.png\" data-fancybox=\"images\" data-fancybox=\"gallery\"><img decoding=\"async\" src=\"https:\/\/17aitech.com\/wp-content\/uploads\/2024\/10\/frc-4eea268e2aa42a494afea7ea5e0ffcac.png\"><\/a><\/p>\n<section><strong>\u8bad\u7ec3\u8bbe\u7f6e\u00a0<\/strong><\/section>\n<section><\/section>\n<section><strong>\u5c06 LLaMA 3.1 \u4ece PyTorch \u79fb\u690d\u5230 JAX\u00a0<\/strong><\/section>\n<section><a href=\"https:\/\/17aitech.com\/wp-content\/uploads\/2024\/10\/frc-527894d452146bd0b4cfde4612533fbf.png\" data-fancybox=\"images\" data-fancybox=\"gallery\"><img decoding=\"async\" src=\"https:\/\/17aitech.com\/wp-content\/uploads\/2024\/10\/frc-527894d452146bd0b4cfde4612533fbf.png\"><\/a><\/section>\n<section>\u6b64\u524d\uff0cNikhil Sonti \u5206\u4eab\u8fc7\u5982\u4f55\u5c06 LLaMA 3.1 \u4ece PyTorch \u79fb\u690d\u5230 JAX\u3002\u4ed6\u6307\u51fa\uff0c\u76ee\u524d 90% \u7684\u5927\u578b\u8bed\u8a00\u6a21\u578b\uff08LLM\uff09\u90fd\u8fd0\u884c\u5728 NVIDIA GPU \u4e0a\uff0c\u4f46\u5b9e\u9645\u4e0a\u8fd8\u6709\u4e00\u4e9b\u540c\u6837\u5f3a\u5927\u4e14\u6027\u4ef7\u6bd4\u66f4\u9ad8\u7684\u66ff\u4ee3\u65b9\u6848\u3002\u4f8b\u5982\uff0c\u5728 Google TPU \u4e0a\u8bad\u7ec3\u548c\u90e8\u7f72 Llama 3.1 \u7684\u6210\u672c\u6bd4 NVIDIA GPU \u4f4e\u7ea6 30%\u3002<\/section>\n<section><\/section>\n<section>\u7136\u800c\uff0c\u652f\u6301\u975e NVIDIA \u786c\u4ef6\u7684\u5f00\u53d1\u5de5\u5177\u8f83\u4e3a\u532e\u4e4f\u3002Sonti \u6700\u521d\u5c1d\u8bd5\u4f7f\u7528 PyTorch XLA \u5728 TPU \u4e0a\u8bad\u7ec3 Llama 3.1\uff0c\u4f46\u8fc7\u7a0b\u5e76\u4e0d\u987a\u5229\u3002XLA \u4e0e PyTorch \u7684\u96c6\u6210\u4e0d\u591f\u5b8c\u5584\uff0c\u7f3a\u5c11\u4e00\u4e9b\u5173\u952e\u7684\u5e93\uff08\u5982 bitsandbytes \u65e0\u6cd5\u6b63\u5e38\u8fd0\u884c\uff09\uff0c\u540c\u65f6\u8fd8\u9047\u5230\u4e86\u4e00\u4e9b\u96be\u4ee5\u89e3\u51b3\u7684 HuggingFace \u9519\u8bef\u3002<\/section>\n<section><\/section>\n<section>\u4e3a\u6b64\uff0c\u4ed6\u51b3\u5b9a\u8c03\u6574\u7b56\u7565\uff0c\u5c06 Llama 3.1 \u4ece PyTorch \u79fb\u690d\u5230 JAX\uff0c\u6210\u529f\u89e3\u51b3\u4e86\u8fd9\u4e9b\u95ee\u9898\u3002Sonti \u8fd8\u5f55\u5236\u4e86\u8be6\u7ec6\u7684\u6559\u7a0b\u89c6\u9891\uff0c\u5e76\u5f00\u6e90\u4e86\u6240\u6709\u4ee3\u7801\uff1a<\/section>\n<p><a href=\"https:\/\/17aitech.com\/wp-content\/uploads\/2024\/10\/frc-f4b9b29000f8b12fc2c1e5dc5972c1c0.png\" data-fancybox=\"images\" data-fancybox=\"gallery\"><img decoding=\"async\" src=\"https:\/\/17aitech.com\/wp-content\/uploads\/2024\/10\/frc-f4b9b29000f8b12fc2c1e5dc5972c1c0.png\"><\/a><\/p>\n<ul>\n<li>\n<section>\u65b9\u6cd5\u6f14\u793a\uff1ahttps:\/\/dub.sh\/felafax-demo<\/section>\n<\/li>\n<li>\n<section>\u4ee3\u7801\u4ed3\u5e93\uff1ahttps:\/\/github.com\/felafax\/felafax<\/section>\n<\/li>\n<\/ul>\n<section><\/section>\n<section><strong>\u52a0\u8f7d\u6a21\u578b\uff0c\u5e76\u628a\u6a21\u578b\u53c2\u6570\u5206\u7247<\/strong><\/section>\n<section>\u00a0<\/section>\n<section>\u5904\u7406\u50cf LLaMA 405B \u8fd9\u6837\u7684\u8d85\u5927\u6a21\u578b\uff0c\u9700\u8981\u5728\u591a\u4e2a\u8bbe\u5907\u4e4b\u95f4\u9ad8\u6548\u5730\u8fdb\u884c\u53c2\u6570\u5206\u7247\u3002\u4ee5\u4e0b\u662f\u5982\u4f55\u901a\u8fc7 JAX \u5b9e\u73b0\u8fd9\u4e00\u70b9\u7684\u3002<\/section>\n<section><\/section>\n<section><strong>\u5728 JAX \u4e2d\u8fdb\u884c\u53c2\u6570\u5206\u7247<\/strong><\/section>\n<section><\/section>\n<section>\u4e3a\u4e86\u5c06\u5de8\u5927\u7684 LLaMA 405B \u6a21\u578b\u9ad8\u6548\u5730\u5206\u5e03\u5230 8 \u5f20 AMD GPU \u4e0a\uff0c\u9700\u8981\u4f7f\u7528 JAX \u7684\u8bbe\u5907\u7f51\u683c\uff08device mesh\uff09\u529f\u80fd\u3002<\/section>\n<section><\/section>\n<section>\u90e8\u7f72\u4ee3\u7801\uff1ahttps:\/\/github.com\/felafax\/felafax\/blob\/e2a96a0e207e1dc70effde099fe33a9e42a7d5cb\/llama3_jax\/trainer_engine\/jax_utils.py#L69<\/section>\n<section><\/section>\n<section>JAX \u7684\u8bbe\u5907\u7f51\u683c\u53ef\u4ee5\u5e2e\u52a9\u6211\u4eec\u628a\u53ef\u7528\u7684\u8bbe\u5907\u7ec4\u7ec7\u6210\u4e00\u4e2a\u7f51\u683c\uff0c\u8ba9\u6211\u4eec\u53ef\u4ee5\u6307\u5b9a\u5982\u4f55\u628a\u6a21\u578b\u7684\u53c2\u6570\u548c\u8ba1\u7b97\u5206\u914d\u5230\u4e0d\u540c\u7684 GPU \u4e0a\u3002<\/section>\n<section><\/section>\n<section>\u5728\u672c\u6587\u7684\u8bbe\u7f6e\u4e2d\uff0c\u9700\u8981\u521b\u5efa\u4e00\u4e2a\u5f62\u72b6\u4e3a\uff081, 8, 1\uff09\u7684\u7f51\u683c\uff0c\u5e76\u5c06\u8f74\u5206\u522b\u547d\u540d\u4e3a\u6570\u636e\u5e76\u884c\uff08dp\uff09\u3001\u5168\u5206\u7247\u6570\u636e\u5e76\u884c\uff08fsdp\uff09\u548c\u6a21\u578b\u5e76\u884c\uff08mp\uff09\u3002\u7136\u540e\uff0c\u4e3a\u6a21\u578b\u7684\u6bcf\u4e2a\u5f20\u91cf\u5b9a\u4e49\u7279\u5b9a\u7684\u5206\u7247\u89c4\u5219\uff0c\u6307\u5b9a\u8fd9\u4e9b\u7ef4\u5ea6\u5982\u4f55\u6cbf\u7740\u8fd9\u4e9b\u7f51\u683c\u8f74\u8fdb\u884c\u5206\u7247\u3002<\/section>\n<section>\n<pre data-lang=\"ini\"><code>DEVICES = jax.devices () <\/code>\r\n<code>DEVICE_COUNT = len (DEVICES) <\/code>\r\n<code>DEVICE_MESH = mesh_utils.create_device_mesh ((1, 8, 1)) <\/code>\r\n<code>MESH = Mesh (devices=DEVICE_MESH, axis_names=(\"dp\", \"fsdp\", \"mp\"))<\/code><\/pre>\n<\/section>\n<section><strong>\u53ef\u89c6\u5316\u5206\u7247<\/strong><\/section>\n<section><\/section>\n<section>\u53ef\u4ee5\u4f7f\u7528\u4ee5\u4e0b\u4ee3\u7801\u6765\u53ef\u89c6\u5316\u5206\u7247\u7ed3\u679c\uff0c\u4ece\u800c\u65b9\u4fbf\u5730\u9a8c\u8bc1\u5206\u7247\u89c4\u5219\u662f\u5426\u6309\u9884\u671f\u5e94\u7528\u3002<\/section>\n<section>\n<pre data-lang=\"css\"><code>jax.debug.visualize_array_sharding\u00a0<\/code><\/pre>\n<\/section>\n<section><strong>\u5206\u7247\u89c4\u5219<\/strong><\/section>\n<section><\/section>\n<section>\u6a21\u578b\u4e0d\u540c\u7ec4\u4ef6\u7684\u5206\u7247\u89c4\u5219\u5982\u4e0b\u6240\u793a\uff1a<\/section>\n<section><\/section>\n<ul>\n<li>\n<section>\u53c2\u6570\u5982\u4f55\u5206\u7247\uff1a<\/section>\n<\/li>\n<\/ul>\n<section><\/section>\n<section>\u53c2\u6570\u8981\u5728 8 \u4e2a GPU \u4e4b\u95f4\u5206\u914d\u3002\u4f8b\u5982\uff0cLM head\uff08lm_head\/kernel\uff09\u5f20\u91cf\u6709\u4e24\u4e2a\u8f74\uff0c\u6309\u7167 PS (&#8220;fsdp&#8221;, &#8220;mp&#8221;) \u8fdb\u884c\u5206\u7247\u3002\u5728\u672c\u4f8b\u4e2d\u662f 8 \u548c 1\uff0c\u56e0\u6b64\u53ef\u4ee5\u770b\u5230\u8be5\u5f20\u91cf\u5728\u7b2c\u4e00\u4e2a\u8f74\u4e0a\u6cbf\u7740 8 \u4e2a GPU \u88ab\u62c6\u5206\u3002<\/section>\n<section><\/section>\n<ul>\n<li>\n<section>Non-Replicated \u53c2\u6570\uff1a<\/section>\n<\/li>\n<\/ul>\n<section><\/section>\n<section>\u6ca1\u6709\u4efb\u4f55\u5206\u7247\u89c4\u8303\u7684\u53c2\u6570\u4f1a\u5728\u6240\u6709\u8bbe\u5907\u4e0a\u8fdb\u884c\u590d\u5236\u3002\u4f8b\u5982\uff0c\u5c42\u5f52\u4e00\u5316\uff08attention_norm\/kernel \u548c ffn_norm\/kernel\uff09\u6ca1\u6709\u8bbe\u7f6e\u5206\u7247\u89c4\u8303\uff0c\u662f PS (None)\u3002<\/section>\n<section><\/section>\n<section><strong>\u5e94\u7528\u5206\u7247\u51fd\u6570<\/strong><\/section>\n<section>\u00a0<\/section>\n<section>\u5728\u52a0\u8f7d\u6a21\u578b\u65f6\uff0c\u4f7f\u7528\u4ee5\u4e0b\u5206\u7247\u51fd\u6570\u9010\u6b65\u5bf9\u6a21\u578b\u6743\u91cd\u8fdb\u884c\u5206\u7247\uff1a<\/section>\n<section>\n<pre data-lang=\"python\"><code>def make_shard_and_gather_fns (partition_specs):<\/code>\r\n<code>    def make_shard_fn (partition_spec):<\/code>\r\n<code>        out_sharding = NamedSharding (mesh, partition_spec)<\/code>\r\n<code>        def shard_fn (tensor):<\/code>\r\n<code>            return jax.device_put (tensor, out_sharding).block_until_ready ()<\/code>\r\n<code>        return shard_fn<\/code><code>\r\n<\/code><code>\r\n<\/code><code>    shard_fns = jax.tree_util.tree_map (make_shard_fn, partition_specs)<\/code>\r\n<code>    return shard_fns<\/code><code>\r\n<\/code><code>\r\n<\/code><code># Create shard functions based on partitioning rules<\/code>\r\n<code>shard_fns = make_shard_and_gather_fns (partitioning_rules)<\/code><\/pre>\n<\/section>\n<section><\/section>\n<section>\u8fd9\u4f7f\u5f97\u6211\u4eec\u80fd\u591f\u5c06\u6bcf\u4e2a\u53c2\u6570\u653e\u7f6e\u5728\u6307\u5b9a\u7684\u8bbe\u5907\u4e0a\uff0c\u5e76\u6309\u7167\u8bbe\u5b9a\u7684\u5206\u7247\u8fdb\u884c\u5904\u7406\u3002<\/section>\n<section><\/section>\n<section><strong>\u5206\u7247\u8bad\u7ec3 Batch<\/strong><\/section>\n<section><\/section>\n<section>\u6700\u521d\uff0c\u8bad\u7ec3 Batch \u662f\u6b63\u5e38\u521b\u5efa\u7684\uff0c\u4f46\u5728\u8f93\u5165\u6a21\u578b\u4e4b\u524d\uff0c\u9700\u8981\u6309\u7167\u4e0b\u9762\u7684\u4ee3\u7801\u5728 GPU \u4e0a\u8fdb\u884c\u5206\u7247\uff1a<\/section>\n<section>\n<pre data-lang=\"makefile\"><code>train_batch = jax.device_put ( train_batch,<\/code>\r\n<code>NamedSharding (self.mesh, PS (\"dp\", \"fsdp\")))<\/code><\/pre>\n<\/section>\n<section><\/section>\n<section>\u5728\u8fd9\u91cc\uff0c\u6211\u4eec\u6307\u5b9a\u8bad\u7ec3 Batch \u5e94\u8be5\u5728 &#8220;dp&#8221; \u548c &#8220;fsdp&#8221; \u8f74\u4e0a\u8fdb\u884c\u5206\u7247\uff0c\u5728\u672c\u4f8b\u4e2d\u5206\u522b\u5bf9\u5e94\u4e8e\u88ab\u5206\u6210 1 \u548c 8 \u4efd\uff0c\u5982\u679c\u628a\u7ed3\u679c\u53ef\u89c6\u5316\u51fa\u6765\uff0c\u5982\u4e0b\u6240\u793a\uff1a<\/section>\n<section><\/section>\n<section>\u5206\u7247\u524d\uff1a<\/section>\n<section><a href=\"https:\/\/17aitech.com\/wp-content\/uploads\/2024\/10\/frc-0522e44df8f290cc4fab35276031cb44.png\" data-fancybox=\"images\" data-fancybox=\"gallery\"><img decoding=\"async\" src=\"https:\/\/17aitech.com\/wp-content\/uploads\/2024\/10\/frc-0522e44df8f290cc4fab35276031cb44.png\"><\/a><\/section>\n<section>\u5728\u8c03\u7528 \u00a0jax.device_put \u4e4b\u540e\uff1a<\/section>\n<section><a href=\"https:\/\/17aitech.com\/wp-content\/uploads\/2024\/10\/frc-157debcab2c374d41ffeab334024942b.png\" data-fancybox=\"images\" data-fancybox=\"gallery\"><img decoding=\"async\" src=\"https:\/\/17aitech.com\/wp-content\/uploads\/2024\/10\/frc-157debcab2c374d41ffeab334024942b.png\"><\/a><\/section>\n<section><strong>\u52a0\u5165 LoRA<\/strong><\/section>\n<section><\/section>\n<section>LoRA \u901a\u8fc7\u5c06\u6743\u91cd\u66f4\u65b0\u5206\u89e3\u4e3a\u4f4e\u79e9\u77e9\u9635\uff0c\u51cf\u5c11\u4e86\u53ef\u8bad\u7ec3\u53c2\u6570\u7684\u6570\u91cf\uff0c\u8fd9\u5bf9\u4e8e\u5fae\u8c03\u5927\u578b\u6a21\u578b\u7279\u522b\u6709\u6548\u3002\u4ee5\u4e0b\u662f\u5728 AMD GPU \u4e0a\u5fae\u8c03 Llama 3.1-405 \u7684 LoRA \u7684\u8981\u70b9\uff1a<\/section>\n<section><\/section>\n<ul>\n<li>\n<section>\u5c06 LoRA \u53c2\u6570\uff08lora_a \u548c lora_b\uff09\u4e0e\u4e3b\u6a21\u578b\u53c2\u6570\u5206\u5f00\u3002<\/section>\n<\/li>\n<li>\n<section>\u4f7f\u7528 jax.lax.stop_gradient (kernel) \u6765\u9632\u6b62\u5bf9\u4e3b\u6a21\u578b\u6743\u91cd\u7684\u66f4\u65b0\u3002<\/section>\n<\/li>\n<li>\n<section>\u4f7f\u7528 lax.dot_general \u8fdb\u884c\u5feb\u901f\u3001\u7cbe\u786e\u63a7\u5236\u7684\u77e9\u9635\u8fd0\u7b97\u3002<\/section>\n<\/li>\n<li>\n<section>LoRA \u8f93\u51fa\u5728\u6dfb\u52a0\u5230\u4e3b\u8f93\u51fa\u4e4b\u524d\u4f1a\u88ab\u7f29\u653e\u4e3a (self.lora_alpha\/self.lora_rank)\u3002<\/section>\n<\/li>\n<\/ul>\n<section><\/section>\n<section><strong>LoRADense \u5c42<\/strong><\/section>\n<section><\/section>\n<section>\u5728\u6b64\u8bbe\u5b9a\u4e00\u4e2a\u81ea\u5b9a\u4e49\u7684 LoRADense \u5c42\uff0c\u8be5\u5c42\u96c6\u6210\u4e86 LoRA \u53c2\u6570\uff1a<\/section>\n<section>\n<pre data-lang=\"ruby\"><code>class LoRADense (nn.Module):<\/code>\r\n<code>    features: int<\/code>\r\n<code>    lora_rank: int = 8<\/code>\r\n<code>    lora_alpha: float = 16.0<\/code>\r\n<code>@nn.compac<\/code><code>t<\/code>\r\n<code>def __call__(self, inputs: Any) -&gt; Any:<\/code>\r\n<code># Original kernel parameter (frozen)<\/code>\r\n<code>        kernel = self.param ('kernel', ...)<\/code>\r\n<code>        y = lax.dot_general (inputs, jax.lax.stop_gradient (kernel), ...)<\/code>\r\n<code># LoRA parameters (trainable)<\/code>\r\n<code>        lora_a = self.variable ('lora_params', 'lora_a', ..., ...)<\/code>\r\n<code>        lora_b = self.variable ('lora_params', 'lora_b', ..., ...)<\/code>\r\n<code># Compute LoRA output<\/code>\r\n<code>        lora_output = lax.dot_general (inputs, lora_a.value, ...)<\/code>\r\n<code>        lora_output = lax.dot_general (lora_output, lora_b.value, ...)<\/code>\r\n<code># Combine original output with LoRA modifications<\/code>\r\n<code>        y += (self.lora_alpha\/self.lora_rank) * lora_output<\/code><code>\r\n<\/code><code>\r\n<\/code><code>        return y.astype (self.dtype)<\/code><\/pre>\n<\/section>\n<section><\/section>\n<section><strong>\u5206\u7247 LoRA \u53c2\u6570<\/strong><\/section>\n<section><\/section>\n<section>\u4e3a\u4e86\u9ad8\u6548\u5730\u5728\u8bbe\u5907\u4e4b\u95f4\u5206\u914d LoRA \u53c2\u6570\uff0c\u6211\u4eec\u4e5f\u901a\u8fc7 JAX \u8bbe\u5b9a\u4e86\u5206\u7247\u89c4\u5219\uff0c\u8fd9\u786e\u4fdd\u4e86 LoRA \u53c2\u6570\u4e0e\u4e3b\u6a21\u578b\u53c2\u6570\u7684\u5206\u7247\u4e00\u81f4\uff0c\u4f18\u5316\u4e86\u5185\u5b58\u4f7f\u7528\u548c\u8ba1\u7b97\u6548\u7387\u3002<\/section>\n<section>\n<pre data-lang=\"nginx\"><code>LoRA A matrices (lora_a)<\/code><\/pre>\n<\/section>\n<section><\/section>\n<section>LoRA A \u77e9\u9635\uff08lora_a\uff09<\/section>\n<section><\/section>\n<ul>\n<li>\n<section>\u5206\u7247\u89c4\u5219\uff1aPS (&#8220;fsdp&#8221;, &#8220;mp&#8221;)<\/section>\n<\/li>\n<li>\n<section>\u53ef\u89c6\u5316\u7ed3\u679c\uff1a\u5982\u4e0b\u56fe\u6240\u793a\uff0clora_a \u53c2\u6570\u88ab\u5206\u7247\u4e3a (8, 1)\uff0c\u8fd9\u610f\u5473\u7740\u7b2c\u4e00\u4e2a\u8f74\u5728 8 \u4e2a\u8bbe\u5907\u4e0a\u8fdb\u884c\u5206\u7247\uff08&#8221;fsdp&#8221; \u8f74\uff09\uff0c\u800c\u7b2c\u4e8c\u4e2a\u8f74\u672a\u8fdb\u884c\u5206\u7247\u3002<\/section>\n<\/li>\n<\/ul>\n<section><a href=\"https:\/\/17aitech.com\/wp-content\/uploads\/2024\/10\/frc-dd9a373097790b3038689656e01940f1.png\" data-fancybox=\"images\" data-fancybox=\"gallery\"><img decoding=\"async\" src=\"https:\/\/17aitech.com\/wp-content\/uploads\/2024\/10\/frc-dd9a373097790b3038689656e01940f1.png\"><\/a><\/section>\n<section><strong>LoRA B \u77e9\u9635\uff08lora_b\uff09<\/strong><\/section>\n<section><\/section>\n<ul>\n<li>\n<section>\u5206\u7247\u89c4\u5219\uff1aPS (&#8220;mp&#8221;, &#8220;fsdp&#8221;)<\/section>\n<\/li>\n<li>\n<section>\u53ef\u89c6\u5316\u7ed3\u679c\uff1a\u5982\u4e0b\u56fe\u6240\u793a\uff0clora_b \u53c2\u6570\u88ab\u5206\u7247\u4e3a (1, 8)\uff0c\u8fd9\u610f\u5473\u7740\u7b2c\u4e8c\u4e2a\u8f74\u5728 8 \u4e2a\u8bbe\u5907\u4e0a\u8fdb\u884c\u5206\u7247\uff08fsdp \u8f74\uff09\uff0c\u800c\u7b2c\u4e00\u4e2a\u8f74\u672a\u8fdb\u884c\u5206\u7247\u3002<\/section>\n<\/li>\n<\/ul>\n<section><a href=\"https:\/\/17aitech.com\/wp-content\/uploads\/2024\/10\/frc-df817e1eb20cbcddf7080c9c9e19add5.png\" data-fancybox=\"images\" data-fancybox=\"gallery\"><img decoding=\"async\" src=\"https:\/\/17aitech.com\/wp-content\/uploads\/2024\/10\/frc-df817e1eb20cbcddf7080c9c9e19add5.png\"><\/a><\/section>\n<section>\u8fd9\u79cd\u5206\u7247\u7b56\u7565\u4f18\u5316\u4e86\u53c2\u6570\u7684\u5206\u914d\uff0c\u51cf\u5c11\u4e86\u901a\u4fe1\u5f00\u9500\uff0c\u5e76\u5728\u8bad\u7ec3\u8fc7\u7a0b\u4e2d\u589e\u5f3a\u4e86\u5e76\u884c\u6027\u3002\u5b83\u786e\u4fdd\u6bcf\u4e2a\u8bbe\u5907\u4ec5\u6301\u6709\u4e00\u90e8\u5206 LoRA \u53c2\u6570\uff0c\u4f7f\u5f97\u5927\u6a21\u578b\u5982 LLaMA 405B \u7684\u9ad8\u6548\u6269\u5c55\u6210\u4e3a\u53ef\u80fd\u3002<\/section>\n<section><\/section>\n<section><strong>\u4ec5\u66f4\u65b0 LoRA \u53c2\u6570\u00a0<\/strong><\/section>\n<section><\/section>\n<section>\u4e3a\u4e86\u4f18\u5316\u8bad\u7ec3\uff0c\u5728\u5fae\u8c03 LLaMA 405B \u6a21\u578b\uff0c\u53ea\u8ba1\u7b97 LoRA \u53c2\u6570\u7684\u68af\u5ea6\uff0c\u4fdd\u6301\u4e3b\u6a21\u578b\u53c2\u6570\u4e0d\u53d8\u3002\u8fd9\u4e2a\u65b9\u6cd5\u51cf\u5c11\u4e86\u5185\u5b58\u4f7f\u7528\uff0c\u5e76\u52a0\u901f\u4e86\u8bad\u7ec3\uff0c\u56e0\u4e3a\u53ea\u66f4\u65b0\u8f83\u5c11\u7684\u53c2\u6570\u3002\u53ef\u4ee5\u79fb\u6b65 GitHub \u4ed3\u5e93\uff0c\u67e5\u770b\u5b9e\u73b0\u7ec6\u8282\u3002<\/section>\n<section><\/section>\n<section>\u5728\u8bad\u7ec3\u8fc7\u7a0b\u4e2d\uff0c\u6bcf\u4e00\u6b65\u90fd\u6d89\u53ca\u5c06\u4e00\u6279\u8f93\u5165\u6570\u636e\u901a\u8fc7\u6a21\u578b\u8fdb\u884c\u5904\u7406\u3002\u7531\u4e8e\u53ea\u6709 LoRA \u53c2\u6570\u662f\u53ef\u8bad\u7ec3\u7684\uff0c\u56e0\u6b64\u6a21\u578b\u7684\u9884\u6d4b\u548c\u8ba1\u7b97\u7684\u635f\u5931\u4ec5\u4f9d\u8d56\u4e8e\u8fd9\u4e9b\u53c2\u6570\uff0c\u7136\u540e\u5bf9 LoRA \u53c2\u6570\u8fdb\u884c\u53cd\u5411\u4f20\u64ad\u3002\u53ea\u66f4\u65b0\u8fd9\u4e9b\u53c2\u6570\u7b80\u5316\u4e86\u8bad\u7ec3\u8fc7\u7a0b\uff0c\u4f7f\u5f97\u5728\u591a\u4e2a GPU \u4e0a\u9ad8\u6548\u5fae\u8c03\u50cf LLaMA 405B \u8fd9\u6837\u7684\u5927\u578b\u6a21\u578b\u6210\u4e3a\u53ef\u80fd\u3002<\/section>\n<section><\/section>\n<section>\u66f4\u591a\u7814\u7a76\u7ec6\u8282\uff0c\u8bf7\u53c2\u8003\u539f\u535a\u5ba2\u3002<\/section>\n<p>\u6587\u7ae0\u6765\u6e90\u4e8e\u4e92\u8054\u7f51:<a href=\"https:\/\/www.jiqizhixin.com\/articles\/2024-10-08-4\" target=\"_blank\">\u5fae\u8c03\u5927\u6a21\u578b\uff0cAMD MI300X\u5c31\u591f\u4e86\uff01\u8ddf\u7740\u8fd9\u7bc7\u535a\u5ba2\u5fae\u8c03Llama 3.1 405B\uff0c\u6548\u679c\u5ab2\u7f8eH100<\/a><\/p>\n","protected":false},"excerpt":{"rendered":"<p>\u6587\u7ae0\u6765\u6e90\u4e8e\u4e92\u8054\u7f51:\u5fae\u8c03\u5927\u6a21\u578b\uff0cAMD M [&hellip;]<\/p>\n","protected":false},"author":3,"featured_media":0,"comment_status":"open","ping_status":"","sticky":false,"template":"","format":"standard","meta":{"site-sidebar-layout":"default","site-content-layout":"","ast-site-content-layout":"","site-content-style":"default","site-sidebar-style":"default","ast-global-header-display":"","ast-banner-title-visibility":"","ast-main-header-display":"","ast-hfb-above-header-display":"","ast-hfb-below-header-display":"","ast-hfb-mobile-header-display":"","site-post-title":"","ast-breadcrumbs-content":"","ast-featured-img":"","footer-sml-layout":"","theme-transparent-header-meta":"","adv-header-id-meta":"","stick-header-meta":"","header-above-stick-meta":"","header-main-stick-meta":"","header-below-stick-meta":"","astra-migrate-meta-layouts":"default","ast-page-background-enabled":"default","ast-page-background-meta":{"desktop":{"background-color":"var(--ast-global-color-4)","background-image":"","background-repeat":"repeat","background-position":"center center","background-size":"auto","background-attachment":"scroll","background-type":"","background-media":"","overlay-type":"","overlay-color":"","overlay-opacity":"","overlay-gradient":""},"tablet":{"background-color":"","background-image":"","background-repeat":"repeat","background-position":"center center","background-size":"auto","background-attachment":"scroll","background-type":"","background-media":"","overlay-type":"","overlay-color":"","overlay-opacity":"","overlay-gradient":""},"mobile":{"background-color":"","background-image":"","background-repeat":"repeat","background-position":"center center","background-size":"auto","background-attachment":"scroll","background-type":"","background-media":"","overlay-type":"","overlay-color":"","overlay-opacity":"","overlay-gradient":""}},"ast-content-background-meta":{"desktop":{"background-color":"var(--ast-global-color-5)","background-image":"","background-repeat":"repeat","background-position":"center center","background-size":"auto","background-attachment":"scroll","background-type":"","background-media":"","overlay-type":"","overlay-color":"","overlay-opacity":"","overlay-gradient":""},"tablet":{"background-color":"var(--ast-global-color-5)","background-image":"","background-repeat":"repeat","background-position":"center center","background-size":"auto","background-attachment":"scroll","background-type":"","background-media":"","overlay-type":"","overlay-color":"","overlay-opacity":"","overlay-gradient":""},"mobile":{"background-color":"var(--ast-global-color-5)","background-image":"","background-repeat":"repeat","background-position":"center center","background-size":"auto","background-attachment":"scroll","background-type":"","background-media":"","overlay-type":"","overlay-color":"","overlay-opacity":"","overlay-gradient":""}},"footnotes":""},"categories":[27],"tags":[52,68,55],"class_list":["post-35955","post","type-post","status-publish","format-standard","hentry","category-news","tag-numpy","tag-68","tag-55"],"yoast_head":"<!-- This site is optimized with the Yoast SEO plugin v26.4 - https:\/\/yoast.com\/wordpress\/plugins\/seo\/ -->\n<title>\u5fae\u8c03\u5927\u6a21\u578b\uff0cAMD MI300X\u5c31\u591f\u4e86\uff01\u8ddf\u7740\u8fd9\u7bc7\u535a\u5ba2\u5fae\u8c03Llama 3.1 405B\uff0c\u6548\u679c\u5ab2\u7f8eH100 - \u4e00\u8d77AI\u6280\u672f<\/title>\n<meta name=\"robots\" content=\"index, follow, max-snippet:-1, max-image-preview:large, max-video-preview:-1\" \/>\n<link rel=\"canonical\" href=\"https:\/\/17aitech.com\/?p=35955\" \/>\n<script type=\"application\/ld+json\" class=\"yoast-schema-graph\">{\"@context\":\"https:\/\/schema.org\",\"@graph\":[{\"@type\":\"WebPage\",\"@id\":\"https:\/\/17aitech.com\/?p=35955\",\"url\":\"https:\/\/17aitech.com\/?p=35955\",\"name\":\"\u5fae\u8c03\u5927\u6a21\u578b\uff0cAMD MI300X\u5c31\u591f\u4e86\uff01\u8ddf\u7740\u8fd9\u7bc7\u535a\u5ba2\u5fae\u8c03Llama 3.1 405B\uff0c\u6548\u679c\u5ab2\u7f8eH100 - \u4e00\u8d77AI\u6280\u672f\",\"isPartOf\":{\"@id\":\"https:\/\/17aitech.com\/#website\"},\"primaryImageOfPage\":{\"@id\":\"https:\/\/17aitech.com\/?p=35955#primaryimage\"},\"image\":{\"@id\":\"https:\/\/17aitech.com\/?p=35955#primaryimage\"},\"thumbnailUrl\":\"https:\/\/17aitech.com\/wp-content\/uploads\/2024\/10\/frc-40479d1ba65cdc53f5e58dc64a913fec.png\",\"datePublished\":\"2025-01-03T06:00:10+00:00\",\"author\":{\"@id\":\"https:\/\/17aitech.com\/#\/schema\/person\/60225458499e817ae0af73e67e440b9d\"},\"breadcrumb\":{\"@id\":\"https:\/\/17aitech.com\/?p=35955#breadcrumb\"},\"inLanguage\":\"zh-Hans\",\"potentialAction\":[{\"@type\":\"ReadAction\",\"target\":[\"https:\/\/17aitech.com\/?p=35955\"]}]},{\"@type\":\"ImageObject\",\"inLanguage\":\"zh-Hans\",\"@id\":\"https:\/\/17aitech.com\/?p=35955#primaryimage\",\"url\":\"https:\/\/17aitech.com\/wp-content\/uploads\/2024\/10\/frc-40479d1ba65cdc53f5e58dc64a913fec.png\",\"contentUrl\":\"https:\/\/17aitech.com\/wp-content\/uploads\/2024\/10\/frc-40479d1ba65cdc53f5e58dc64a913fec.png\"},{\"@type\":\"BreadcrumbList\",\"@id\":\"https:\/\/17aitech.com\/?p=35955#breadcrumb\",\"itemListElement\":[{\"@type\":\"ListItem\",\"position\":1,\"name\":\"\u9996\u9875\",\"item\":\"https:\/\/17aitech.com\/\"},{\"@type\":\"ListItem\",\"position\":2,\"name\":\"\u5fae\u8c03\u5927\u6a21\u578b\uff0cAMD MI300X\u5c31\u591f\u4e86\uff01\u8ddf\u7740\u8fd9\u7bc7\u535a\u5ba2\u5fae\u8c03Llama 3.1 405B\uff0c\u6548\u679c\u5ab2\u7f8eH100\"}]},{\"@type\":\"WebSite\",\"@id\":\"https:\/\/17aitech.com\/#website\",\"url\":\"https:\/\/17aitech.com\/\",\"name\":\"\u4e00\u8d77AI\u6280\u672f\",\"description\":\"\u8ba9AI\u77e5\u8bc6\u89e6\u624b\u53ef\u53ca\",\"alternateName\":\"\u4e00\u8d77AI\u6280\u672f\",\"potentialAction\":[{\"@type\":\"SearchAction\",\"target\":{\"@type\":\"EntryPoint\",\"urlTemplate\":\"https:\/\/17aitech.com\/?s={search_term_string}\"},\"query-input\":{\"@type\":\"PropertyValueSpecification\",\"valueRequired\":true,\"valueName\":\"search_term_string\"}}],\"inLanguage\":\"zh-Hans\"},{\"@type\":\"Person\",\"@id\":\"https:\/\/17aitech.com\/#\/schema\/person\/60225458499e817ae0af73e67e440b9d\",\"name\":\"AI\u5c0f\u52a9\u624b\",\"image\":{\"@type\":\"ImageObject\",\"inLanguage\":\"zh-Hans\",\"@id\":\"https:\/\/17aitech.com\/#\/schema\/person\/image\/\",\"url\":\"\/\/17aitech.com\/wp-content\/uploads\/2024\/04\/robot_3.png\",\"contentUrl\":\"\/\/17aitech.com\/wp-content\/uploads\/2024\/04\/robot_3.png\",\"caption\":\"AI\u5c0f\u52a9\u624b\"},\"description\":\"\u8fd9\u4e2a\u4eba\u5f88\u61d2\uff0c\u4ec0\u4e48\u90fd\u6ca1\u6709\u7559\u4e0b\uff5e\",\"url\":\"https:\/\/17aitech.com\/?page_id=33738&user=3\"}]}<\/script>\n<!-- \/ Yoast SEO plugin. -->","yoast_head_json":{"title":"\u5fae\u8c03\u5927\u6a21\u578b\uff0cAMD MI300X\u5c31\u591f\u4e86\uff01\u8ddf\u7740\u8fd9\u7bc7\u535a\u5ba2\u5fae\u8c03Llama 3.1 405B\uff0c\u6548\u679c\u5ab2\u7f8eH100 - \u4e00\u8d77AI\u6280\u672f","robots":{"index":"index","follow":"follow","max-snippet":"max-snippet:-1","max-image-preview":"max-image-preview:large","max-video-preview":"max-video-preview:-1"},"canonical":"https:\/\/17aitech.com\/?p=35955","schema":{"@context":"https:\/\/schema.org","@graph":[{"@type":"WebPage","@id":"https:\/\/17aitech.com\/?p=35955","url":"https:\/\/17aitech.com\/?p=35955","name":"\u5fae\u8c03\u5927\u6a21\u578b\uff0cAMD MI300X\u5c31\u591f\u4e86\uff01\u8ddf\u7740\u8fd9\u7bc7\u535a\u5ba2\u5fae\u8c03Llama 3.1 405B\uff0c\u6548\u679c\u5ab2\u7f8eH100 - \u4e00\u8d77AI\u6280\u672f","isPartOf":{"@id":"https:\/\/17aitech.com\/#website"},"primaryImageOfPage":{"@id":"https:\/\/17aitech.com\/?p=35955#primaryimage"},"image":{"@id":"https:\/\/17aitech.com\/?p=35955#primaryimage"},"thumbnailUrl":"https:\/\/17aitech.com\/wp-content\/uploads\/2024\/10\/frc-40479d1ba65cdc53f5e58dc64a913fec.png","datePublished":"2025-01-03T06:00:10+00:00","author":{"@id":"https:\/\/17aitech.com\/#\/schema\/person\/60225458499e817ae0af73e67e440b9d"},"breadcrumb":{"@id":"https:\/\/17aitech.com\/?p=35955#breadcrumb"},"inLanguage":"zh-Hans","potentialAction":[{"@type":"ReadAction","target":["https:\/\/17aitech.com\/?p=35955"]}]},{"@type":"ImageObject","inLanguage":"zh-Hans","@id":"https:\/\/17aitech.com\/?p=35955#primaryimage","url":"https:\/\/17aitech.com\/wp-content\/uploads\/2024\/10\/frc-40479d1ba65cdc53f5e58dc64a913fec.png","contentUrl":"https:\/\/17aitech.com\/wp-content\/uploads\/2024\/10\/frc-40479d1ba65cdc53f5e58dc64a913fec.png"},{"@type":"BreadcrumbList","@id":"https:\/\/17aitech.com\/?p=35955#breadcrumb","itemListElement":[{"@type":"ListItem","position":1,"name":"\u9996\u9875","item":"https:\/\/17aitech.com\/"},{"@type":"ListItem","position":2,"name":"\u5fae\u8c03\u5927\u6a21\u578b\uff0cAMD MI300X\u5c31\u591f\u4e86\uff01\u8ddf\u7740\u8fd9\u7bc7\u535a\u5ba2\u5fae\u8c03Llama 3.1 405B\uff0c\u6548\u679c\u5ab2\u7f8eH100"}]},{"@type":"WebSite","@id":"https:\/\/17aitech.com\/#website","url":"https:\/\/17aitech.com\/","name":"\u4e00\u8d77AI\u6280\u672f","description":"\u8ba9AI\u77e5\u8bc6\u89e6\u624b\u53ef\u53ca","alternateName":"\u4e00\u8d77AI\u6280\u672f","potentialAction":[{"@type":"SearchAction","target":{"@type":"EntryPoint","urlTemplate":"https:\/\/17aitech.com\/?s={search_term_string}"},"query-input":{"@type":"PropertyValueSpecification","valueRequired":true,"valueName":"search_term_string"}}],"inLanguage":"zh-Hans"},{"@type":"Person","@id":"https:\/\/17aitech.com\/#\/schema\/person\/60225458499e817ae0af73e67e440b9d","name":"AI\u5c0f\u52a9\u624b","image":{"@type":"ImageObject","inLanguage":"zh-Hans","@id":"https:\/\/17aitech.com\/#\/schema\/person\/image\/","url":"\/\/17aitech.com\/wp-content\/uploads\/2024\/04\/robot_3.png","contentUrl":"\/\/17aitech.com\/wp-content\/uploads\/2024\/04\/robot_3.png","caption":"AI\u5c0f\u52a9\u624b"},"description":"\u8fd9\u4e2a\u4eba\u5f88\u61d2\uff0c\u4ec0\u4e48\u90fd\u6ca1\u6709\u7559\u4e0b\uff5e","url":"https:\/\/17aitech.com\/?page_id=33738&user=3"}]}},"_links":{"self":[{"href":"https:\/\/17aitech.com\/index.php?rest_route=\/wp\/v2\/posts\/35955","targetHints":{"allow":["GET"]}}],"collection":[{"href":"https:\/\/17aitech.com\/index.php?rest_route=\/wp\/v2\/posts"}],"about":[{"href":"https:\/\/17aitech.com\/index.php?rest_route=\/wp\/v2\/types\/post"}],"author":[{"embeddable":true,"href":"https:\/\/17aitech.com\/index.php?rest_route=\/wp\/v2\/users\/3"}],"replies":[{"embeddable":true,"href":"https:\/\/17aitech.com\/index.php?rest_route=%2Fwp%2Fv2%2Fcomments&post=35955"}],"version-history":[{"count":0,"href":"https:\/\/17aitech.com\/index.php?rest_route=\/wp\/v2\/posts\/35955\/revisions"}],"wp:attachment":[{"href":"https:\/\/17aitech.com\/index.php?rest_route=%2Fwp%2Fv2%2Fmedia&parent=35955"}],"wp:term":[{"taxonomy":"category","embeddable":true,"href":"https:\/\/17aitech.com\/index.php?rest_route=%2Fwp%2Fv2%2Fcategories&post=35955"},{"taxonomy":"post_tag","embeddable":true,"href":"https:\/\/17aitech.com\/index.php?rest_route=%2Fwp%2Fv2%2Ftags&post=35955"}],"curies":[{"name":"wp","href":"https:\/\/api.w.org\/{rel}","templated":true}]}}