{"id":3701,"date":"2024-05-11T13:33:43","date_gmt":"2024-05-11T05:33:43","guid":{"rendered":"https:\/\/www.aqwu.net\/wp\/?p=3701"},"modified":"2024-10-21T08:27:27","modified_gmt":"2024-10-21T00:27:27","slug":"ai-%e5%bc%80%e5%8f%91%e7%9a%84%e4%b8%80%e4%ba%9b%e5%b8%b8%e7%94%a8%e7%9f%a5%e8%af%86","status":"publish","type":"post","link":"https:\/\/www.aqwu.net\/wp\/?p=3701","title":{"rendered":"AI \u5f00\u53d1\u7684\u4e00\u4e9b\u5e38\u7528\u77e5\u8bc6"},"content":{"rendered":"\n<h2 class=\"wp-block-heading\"><strong>1. HF \u76f8\u5173\u7684\u73af\u5883\u53d8\u91cf<\/strong><\/h2>\n\n\n\n<p>HF\uff08Hugging Face\uff09\u5e93\u4f7f\u7528\u4e86\u591a\u4e2a\u73af\u5883\u53d8\u91cf\u6765\u914d\u7f6e\u5176\u884c\u4e3a\uff0c\u7279\u522b\u662f\u5728\u6a21\u578b\u4e0b\u8f7d\u548c\u7f13\u5b58\u7ba1\u7406\u65b9\u9762\u3002\u4ee5\u4e0b\u662f\u4e00\u4e9b\u5e38\u89c1\u7684Hugging Face\u76f8\u5173\u73af\u5883\u53d8\u91cf\uff1a<\/p>\n\n\n\n<ol class=\"wp-block-list\">\n<li><strong>TRANSFORMERS_CACHE<\/strong>\uff1a\u7528\u4e8e\u8bbe\u7f6eTransformers\u5e93\u6a21\u578b\u548c\u914d\u7f6e\u6587\u4ef6\u7684\u7f13\u5b58\u76ee\u5f55\u3002<\/li>\n\n\n\n<li><strong>HF_HOME<\/strong>\uff1a\u8bbe\u7f6eHugging Face\u5e93\u7684\u4e3b\u76ee\u5f55\uff0c\u7528\u4e8e\u5b58\u653e\u6a21\u578b\u548ctokenizers\u7684\u7f13\u5b58\u3002<\/li>\n\n\n\n<li><strong>HF_DATASETS_CACHE<\/strong>\uff1a\u7528\u4e8e\u8bbe\u7f6eDatasets\u5e93\u7684\u7f13\u5b58\u76ee\u5f55\u3002<\/li>\n\n\n\n<li><strong>HF_METRICS_CACHE<\/strong>\uff1a\u7528\u4e8e\u8bbe\u7f6eMetrics\u5e93\u7684\u7f13\u5b58\u76ee\u5f55\u3002<\/li>\n\n\n\n<li><strong>HF_MODULES_CACHE<\/strong>\uff1a\u7528\u4e8e\u8bbe\u7f6eHugging Face Hub\u7684\u6a21\u5757\u7f13\u5b58\u76ee\u5f55\u3002<\/li>\n\n\n\n<li><strong>TRANSFORMERS_VERBOSITY<\/strong>\uff1a\u7528\u6765\u8bbe\u7f6e\u5e93\u7684\u65e5\u5fd7\u7ea7\u522b\uff0c\u5982&#8221;info&#8221;\u3001&#8221;error&#8221;\u3001&#8221;warning&#8221;\u7b49\u3002<\/li>\n\n\n\n<li><strong>HF_HUB_CACHE<\/strong>\uff1a\u7528\u4e8e\u8bbe\u7f6e\u6a21\u578b\u548c\u5176\u4ed6\u6587\u4ef6\u7684\u9ed8\u8ba4\u7f13\u5b58\u4f4d\u7f6e\u3002<\/li>\n\n\n\n<li><strong>HF_HUB_DISABLE_OUTDATED_WARNING<\/strong>\uff1a\u8bbe\u7f6e\u4e3a&#8221;1&#8243;\u53ef\u4ee5\u7981\u7528\u8fc7\u65f6\u8b66\u544a\u3002<\/li>\n\n\n\n<li><strong>TRANSFORMERS_OFFLINE<\/strong>\uff1a\u8bbe\u7f6e\u4e3a&#8221;1&#8243;\u540e\uff0cTransformers\u5e93\u5c06\u5728\u79bb\u7ebf\u6a21\u5f0f\u4e0b\u8fd0\u884c\uff0c\u53ea\u4f7f\u7528\u672c\u5730\u7f13\u5b58\u7684\u6587\u4ef6\u3002<\/li>\n\n\n\n<li><strong>HF_HUB_TOKEN<\/strong>\uff1a\u7528\u4e8e\u8ba4\u8bc1\u548c\u4e0b\u8f7d\u9700\u8981\u6743\u9650\u7684\u6a21\u578b\u3002<\/li>\n\n\n\n<li><strong>HF_ENDPOINT<\/strong>: \u7528\u4e8e\u6307\u5b9a\u4e0e Hugging Face Hub \u8fde\u63a5\u7684\u81ea\u5b9a\u4e49\u670d\u52a1\u5668\u5730\u5740<\/li>\n<\/ol>\n\n\n\n<p>\u8fd9\u4e9b\u73af\u5883\u53d8\u91cf\u53ef\u4ee5\u5e2e\u52a9\u7528\u6237\u81ea\u5b9a\u4e49\u548c\u4f18\u5316\u4ed6\u4eec\u7684\u5de5\u4f5c\u6d41\u7a0b\uff0c\u7279\u522b\u662f\u5728\u5904\u7406\u5927\u91cf\u6570\u636e\u548c\u6a21\u578b\u65f6\uff0c\u6216\u8005\u5728\u6709\u9650\u7684\u7f51\u7edc\u8fde\u63a5\u73af\u5883\u4e0b\u5de5\u4f5c\u3002\u901a\u8fc7\u8bbe\u7f6e\u8fd9\u4e9b\u73af\u5883\u53d8\u91cf\uff0c\u7528\u6237\u53ef\u4ee5\u63a7\u5236\u6a21\u578b\u548c\u6570\u636e\u7684\u4e0b\u8f7d\u3001\u7f13\u5b58\u548c\u5b58\u50a8\u4f4d\u7f6e\u3002<\/p>\n\n\n\n<p>\u8bbe\u7f6e\u73af\u5883\u53d8\u91cf\u7684\u65b9\u6cd5\u53d6\u51b3\u4e8e\u4f60\u4f7f\u7528\u7684\u64cd\u4f5c\u7cfb\u7edf\u3002\u4e0b\u9762\u6211\u5c06\u4e3aWindows\u3001Linux\u548cmacOS\u64cd\u4f5c\u7cfb\u7edf\u63d0\u4f9b\u4e00\u4e9b\u793a\u4f8b\u3002<\/p>\n\n\n\n<h3 class=\"wp-block-heading\"><strong>1.1 \u5bf9\u4e8eWindows<\/strong><\/h3>\n\n\n\n<p>\u5728Windows\u7cfb\u7edf\u4e2d\uff0c\u4f60\u53ef\u4ee5\u4f7f\u7528\u547d\u4ee4\u63d0\u793a\u7b26\u6216PowerShell\u6765\u8bbe\u7f6e\u73af\u5883\u53d8\u91cf\u3002\u4f8b\u5982\uff0c\u8bbe\u7f6e<code>TRANSFORMERS_CACHE<\/code>\uff1a<\/p>\n\n\n\n<p><strong>\u547d\u4ee4\u63d0\u793a\u7b26<\/strong>:<\/p>\n\n\n\n<div class=\"wp-block-urvanov-syntax-highlighter-code-block\"><pre class=\"lang:sh decode:true \">set TRANSFORMERS_CACHE=C:\\path\\to\\your\\cache<\/pre><\/div>\n\n\n\n<p><strong>PowerShell<\/strong>:<\/p>\n\n\n\n<div class=\"wp-block-urvanov-syntax-highlighter-code-block\"><pre class=\"lang:python decode:true \">$env:TRANSFORMERS_CACHE=\"C:\\path\\to\\your\\cache\"\n<\/pre><\/div>\n\n\n\n<p>\u8fd9\u4e9b\u547d\u4ee4\u53ea\u4f1a\u5728\u5f53\u524d\u7684\u547d\u4ee4\u884c\u4f1a\u8bdd\u4e2d\u6709\u6548\u3002\u82e5\u8981\u6c38\u4e45\u8bbe\u7f6e\u73af\u5883\u53d8\u91cf\uff0c\u4f60\u53ef\u4ee5\u5728\u7cfb\u7edf\u7684\u201c\u73af\u5883\u53d8\u91cf\u201d\u8bbe\u7f6e\u4e2d\u6dfb\u52a0\u5b83\u4eec\u3002<\/p>\n\n\n\n<h3 class=\"wp-block-heading\"><strong>1.2 \u5bf9\u4e8eLinux\u548cmacOS<\/strong><\/h3>\n\n\n\n<p>\u5728Linux\u6216macOS\u7cfb\u7edf\u4e2d\uff0c\u4f60\u53ef\u4ee5\u5728shell\u4e2d\u4f7f\u7528<code>export<\/code>\u547d\u4ee4\u8bbe\u7f6e\u73af\u5883\u53d8\u91cf\u3002\u8fd9\u53ef\u4ee5\u5728\u7ec8\u7aef\u4e2d\u4e34\u65f6\u8bbe\u7f6e\uff0c\u6216\u8005\u6dfb\u52a0\u5230\u4f60\u7684<code>~\/.bashrc<\/code>\u3001<code>~\/.zshrc<\/code>\u6216\u5176\u4ed6shell\u914d\u7f6e\u6587\u4ef6\u4e2d\uff0c\u4ee5\u5b9e\u73b0\u6301\u4e45\u8bbe\u7f6e\u3002<\/p>\n\n\n\n<div class=\"wp-block-urvanov-syntax-highlighter-code-block\"><pre class=\"lang:sh decode:true \">export TRANSFORMERS_CACHE=\/path\/to\/your\/cache\n<\/pre><\/div>\n\n\n\n<p>\u6dfb\u52a0\u5230\u914d\u7f6e\u6587\u4ef6\u540e\uff0c\u6bcf\u6b21\u6253\u5f00\u65b0\u7684\u7ec8\u7aef\u7a97\u53e3\u65f6\uff0c\u73af\u5883\u53d8\u91cf\u5c31\u4f1a\u81ea\u52a8\u8bbe\u7f6e\u3002<\/p>\n\n\n\n<h3 class=\"wp-block-heading\"><strong>1.3 \u5b9e\u7528\u793a\u4f8b<\/strong><\/h3>\n\n\n\n<h4 class=\"wp-block-heading\"><strong>1.3.1 \u8bbe\u7f6e\u6a21\u578b\u7f13\u5b58\u4f4d\u7f6e<\/strong><\/h4>\n\n\n\n<div class=\"wp-block-urvanov-syntax-highlighter-code-block\"><pre class=\"lang:sh decode:true \">export TRANSFORMERS_CACHE=\/home\/user\/huggingface\/transformers\/<\/pre><\/div>\n\n\n\n<h4 class=\"wp-block-heading\"><strong>1.3.2\u8bbe\u7f6e\u6570\u636e\u96c6\u7f13\u5b58\u4f4d\u7f6e<\/strong><\/h4>\n\n\n\n<div class=\"wp-block-urvanov-syntax-highlighter-code-block\"><pre class=\"lang:sh decode:true \">export HF_DATASETS_CACHE=\/home\/user\/huggingface\/datasets\/<\/pre><\/div>\n\n\n\n<h4 class=\"wp-block-heading\"><strong>1.3.3 \u5728\u79bb\u7ebf\u6a21\u5f0f\u4e0b\u8fd0\u884cTransformers<\/strong><\/h4>\n\n\n\n<div class=\"wp-block-urvanov-syntax-highlighter-code-block\"><pre class=\"lang:sh decode:true \">export TRANSFORMERS_OFFLINE=1<\/pre><\/div>\n\n\n\n<h4 class=\"wp-block-heading\"><strong>1.3.4 \u8bbe\u7f6e\u65e5\u5fd7\u7ea7\u522b\u4e3a\u8b66\u544a<\/strong><\/h4>\n\n\n\n<div class=\"wp-block-urvanov-syntax-highlighter-code-block\"><pre class=\"lang:sh decode:true \">export TRANSFORMERS_VERBOSITY=warning<\/pre><\/div>\n\n\n\n<h4 class=\"wp-block-heading\"><strong>1.3.5 \u81ea\u5b9a\u4e49\u670d\u52a1\u5668\u5730\u5740<\/strong><\/h4>\n\n\n\n<p>\u53ef\u4ee5\u8bbe\u7f6e\u56fd\u5185\u7684\u955c\u50cf\u7f51\u7ad9<\/p>\n\n\n\n<div class=\"wp-block-urvanov-syntax-highlighter-code-block\"><pre class=\"lang:sh decode:true \">export HF_ENDPOINT=https:\/\/hf-mirror.com<\/pre><\/div>\n\n\n\n<p>\u901a\u8fc7\u8fd9\u4e9b\u793a\u4f8b\uff0c\u4f60\u53ef\u4ee5\u6839\u636e\u81ea\u5df1\u7684\u9700\u6c42\u914d\u7f6eHugging Face\u5e93\u7684\u884c\u4e3a\uff0c\u4f18\u5316\u672c\u5730\u5f00\u53d1\u73af\u5883\u3002<\/p>\n\n\n\n<h3 class=\"wp-block-heading\"><strong>1.3.6 \u4f7f\u7528 huggingface-cli \u4e0b\u8f7d<\/strong><\/h3>\n\n\n\n<div class=\"wp-block-urvanov-syntax-highlighter-code-block\"><pre class=\"lang:sh decode:true \">pip install transformers\nhuggingface-cli login\n\nhuggingface-cli download Qwen\/Qwen2-VL-7B-Instruct-GPTQ-Int4 --local-dir .\/Qwen\/Qwen2-VL-7B-Instruct-GPTQ-Int4 --local-dir-use-symlinks False\n<\/pre><\/div>\n\n\n\n<p>\u5982\u679c\u9700\u8981\u4e0b\u8f7d\u6570\u636e\u96c6\uff0c\u5219\u9700\u8981\u5728download \u540e\u9762\u52a0\u4e0a &#8211;repo-type dataset,\u4f8b\u5982:<\/p>\n\n\n\n<div class=\"wp-block-urvanov-syntax-highlighter-code-block\"><pre class=\"lang:python decode:true \" >huggingface-cli download  --repo-type dataset Idavidrein\/gpqa --local-dir .\/Idavidrein\/gpqa\n<\/pre><\/div>\n\n\n\n<p>\u5982\u679c\u4e0d\u60f3\u767b\u5f55\uff0c\u53ef\u4ee5\u5728\u540e\u9762\u52a0\u4e0a &#8211;token,\u4f8b\u5982:<\/p>\n\n\n\n<div class=\"wp-block-urvanov-syntax-highlighter-code-block\"><pre class=\"lang:python decode:true \" >huggingface-cli download meta-llama\/Meta-Llama-3-8B-Instruct --local-dir .\/meta-llama\/Meta-Llama-3-8B-Instruct --token hf_xxxxxx\n<\/pre><\/div>\n\n\n\n<h2 class=\"wp-block-heading\"><strong>2. \u5e38\u7528\u7684 PyPI \u955c\u50cf\u6e90<\/strong><\/h2>\n\n\n\n<p>\u5728 pip \u5b89\u88c5\u7684\u65f6\u5019\uff0c\u53ef\u4ee5\u4f7f\u7528 pip \u7684 -i \u53c2\u6570<\/p>\n\n\n\n<div class=\"wp-block-urvanov-syntax-highlighter-code-block\"><pre class=\"lang:sh decode:true \">pip install -r requirements.txt -i https:\/\/pypi.org\/simple\/\n<\/pre><\/div>\n\n\n\n<p>\u539f\u751f\u5730\u5740<\/p>\n\n\n\n<div class=\"wp-block-urvanov-syntax-highlighter-code-block\"><pre class=\"lang:sh decode:true \">https:\/\/pypi.org\/simple\/<\/pre><\/div>\n\n\n\n<p>\u6e05\u534e\u5927\u5b66<\/p>\n\n\n\n<div class=\"wp-block-urvanov-syntax-highlighter-code-block\"><pre class=\"lang:sh decode:true \">https:\/\/pypi.tuna.tsinghua.edu.cn\/simple<\/pre><\/div>\n\n\n\n<p>\u963f\u91cc\u4e91<\/p>\n\n\n\n<div class=\"wp-block-urvanov-syntax-highlighter-code-block\"><pre class=\"lang:sh decode:true \">https:\/\/mirrors.aliyun.com\/pypi\/simple\/<\/pre><\/div>\n\n\n\n<p>\u4e2d\u56fd\u79d1\u6280\u5927\u5b66<\/p>\n\n\n\n<div class=\"wp-block-urvanov-syntax-highlighter-code-block\"><pre class=\"lang:sh decode:true \">https:\/\/pypi.mirrors.ustc.edu.cn\/simple\/<\/pre><\/div>\n\n\n\n<p>\u8c46\u74e3(douban)<\/p>\n\n\n\n<div class=\"wp-block-urvanov-syntax-highlighter-code-block\"><pre class=\"lang:sh decode:true \">https:\/\/pypi.doubanio.com\/simple\/<\/pre><\/div>\n\n\n\n<h2 class=\"wp-block-heading\"><strong>3. \u8bbe\u7f6e CUDA_VISIBLE_DEVICES \u53c2\u6570<\/strong><\/h2>\n\n\n\n<p><code>CUDA_VISIBLE_DEVICES<\/code> \u662f\u4e00\u4e2a\u73af\u5883\u53d8\u91cf\uff0c\u7528\u4e8e\u63a7\u5236 CUDA \u5e94\u7528\u7a0b\u5e8f\uff08\u5982 TensorFlow \u6216 PyTorch \u4f7f\u7528\u7684\u5e94\u7528\uff09\u53ef\u89c1\u7684 GPU \u8bbe\u5907\u3002\u901a\u8fc7\u8bbe\u7f6e\u8fd9\u4e2a\u73af\u5883\u53d8\u91cf\uff0c\u4f60\u53ef\u4ee5\u6307\u5b9a\u54ea\u4e9b GPU \u8bbe\u5907\u5e94\u8be5\u88ab\u5e94\u7528\u7a0b\u5e8f\u8bc6\u522b\u548c\u4f7f\u7528\u3002\u8fd9\u662f\u5728\u591aGPU\u7cfb\u7edf\u4e0a\u8fdb\u884c\u8d44\u6e90\u7ba1\u7406\u548c\u9694\u79bb\u7684\u4e00\u79cd\u6709\u6548\u65b9\u5f0f\u3002<\/p>\n\n\n\n<p><strong>\u4f7f\u7528\u5355\u4e2aGPU<\/strong>\uff1a<\/p>\n\n\n\n<div class=\"wp-block-urvanov-syntax-highlighter-code-block\"><pre class=\"lang:sh decode:true \">export CUDA_VISIBLE_DEVICES=0\n<\/pre><\/div>\n\n\n\n<p>\u8fd9\u4e2a\u8bbe\u7f6e\u4f7f\u5f97\u53ea\u6709\u7f16\u53f7\u4e3a 0 \u7684 GPU \u5bf9 CUDA \u5e94\u7528\u7a0b\u5e8f\u53ef\u89c1\u3002<\/p>\n\n\n\n<p><strong>\u6307\u5b9a\u591a\u4e2aGPU<\/strong>\uff1a<\/p>\n\n\n\n<div class=\"wp-block-urvanov-syntax-highlighter-code-block\"><pre class=\"lang:sh decode:true \">export CUDA_VISIBLE_DEVICES=0,1,2\n<\/pre><\/div>\n\n\n\n<p>\u8fd9\u5c06\u4f7f\u5e94\u7528\u7a0b\u5e8f\u80fd\u770b\u5230\u7f16\u53f7\u4e3a 0\u30011 \u548c 2 \u7684 GPU\u3002<\/p>\n\n\n\n<p><strong>\u7981\u7528\u6240\u6709GPU<\/strong><\/p>\n\n\n\n<div class=\"wp-block-urvanov-syntax-highlighter-code-block\"><pre class=\"lang:sh decode:true \">export CUDA_VISIBLE_DEVICES=\"\"\n<\/pre><\/div>\n\n\n\n<p>\u8fd9\u5c06\u5bfc\u81f4\u6ca1\u6709 GPU \u88ab CUDA \u5e94\u7528\u7a0b\u5e8f\u68c0\u6d4b\u5230\uff0c\u53ef\u4ee5\u7528\u6765\u6d4b\u8bd5 CPU-only \u7684\u4ee3\u7801\u8def\u5f84\u3002<\/p>\n\n\n\n<h2 class=\"wp-block-heading\"><strong>4. \u8bbe\u7f6eCUDA\u7684\u52a8\u6001\u5e93\u8def\u5f84\u548c\u8fd0\u884c\u8def\u5f84<\/strong><\/h2>\n\n\n\n<p>\u5728Linux \u73af\u5883\u4e0b\u53ef\u80fd\u9700\u8981\u8bbe\u7f6e cuda \u7684\u4e00\u4e9b\u8def\u5f84<\/p>\n\n\n\n<div class=\"wp-block-urvanov-syntax-highlighter-code-block\"><pre class=\"lang:sh decode:true \">export PATH=\/usr\/local\/cuda-12.4\/bin:$PATH\nexport LD_LIBRARY_PATH=\/usr\/local\/cuda-12.4\/lib64:\/usr\/local\/cuda-12.4\/extras\/CUPTI\/lib64:$LD_LIBRARY_PATH\n<\/pre><\/div>\n\n\n\n<h2 class=\"wp-block-heading\"><strong>5. \u67e5\u770b torch \u7248\u672c<\/strong><\/h2>\n\n\n\n<div class=\"wp-block-urvanov-syntax-highlighter-code-block\"><pre class=\"lang:sh decode:true \">python -c \"import torch; print(torch.__version__)\"<\/pre><\/div>\n\n\n\n<p>\u5982\u679c\u662fCUDA\u7248\u672c\uff0c\u663e\u793a 2.3.0+cu121<\/p>\n\n\n\n<p><strong>\u67e5\u770b cuda \u662f\u5426\u6709\u6548<\/strong><\/p>\n\n\n\n<div class=\"wp-block-urvanov-syntax-highlighter-code-block\"><pre class=\"lang:sh decode:true \">python -c \"import torch; print(torch.cuda.is_available())\"<\/pre><\/div>\n\n\n\n<p>\u5982\u679c\u6709\u6548\uff0c\u5219\u8fd4\u56de True<\/p>\n\n\n\n<h2 class=\"wp-block-heading\"><strong>6. torch \u7684\u5b89\u88c5\u65b9\u6cd5<\/strong><\/h2>\n\n\n\n<p>\u53c2\u7167 <a href=\"https:\/\/pytorch.org\/\">https:\/\/pytorch.org\/<\/a><\/p>\n\n\n\n<h3 class=\"wp-block-heading\"><strong>6.1 Linux + Python + Conda<\/strong><\/h3>\n\n\n\n<h4 class=\"wp-block-heading\"><strong>6.1.0 Conda + CPU<\/strong><\/h4>\n\n\n\n<div class=\"wp-block-urvanov-syntax-highlighter-code-block\"><pre class=\"lang:sh decode:true \">conda install pytorch torchvision torchaudio cpuonly -c pytorch<\/pre><\/div>\n\n\n\n<h4 class=\"wp-block-heading\"><strong>6.1.1 Conda + CUDA11.8<\/strong><\/h4>\n\n\n\n<div class=\"wp-block-urvanov-syntax-highlighter-code-block\"><pre class=\"lang:sh decode:true \">conda install pytorch torchvision torchaudio pytorch-cuda=11.8 -c pytorch -c nvidia<\/pre><\/div>\n\n\n\n<h4 class=\"wp-block-heading\"><strong>6.1.2 Conda + CUDA12.1<\/strong><\/h4>\n\n\n\n<div class=\"wp-block-urvanov-syntax-highlighter-code-block\"><pre class=\"lang:python decode:true \">conda install pytorch torchvision torchaudio pytorch-cuda=12.1 -c pytorch -c nvidia<\/pre><\/div>\n\n\n\n<h4 class=\"wp-block-heading\"><strong>6.1.3 Conda + CUDA12.4<\/strong><\/h4>\n\n\n\n<div class=\"wp-block-urvanov-syntax-highlighter-code-block\"><pre class=\"lang:python decode:true \" >conda install pytorch torchvision torchaudio pytorch-cuda=12.4 -c pytorch -c nvidia<\/pre><\/div>\n\n\n\n<h3 class=\"wp-block-heading\"><strong>6.2 Linux + Python + pip<\/strong><\/h3>\n\n\n\n<h4 class=\"wp-block-heading\"><strong>6.2.0 Pip + CPU<\/strong><\/h4>\n\n\n\n<div class=\"wp-block-urvanov-syntax-highlighter-code-block\"><pre class=\"lang:sh decode:true \">pip3 install torch torchvision torchaudio --index-url https:\/\/download.pytorch.org\/whl\/cpu<\/pre><\/div>\n\n\n\n<h4 class=\"wp-block-heading\"><strong>6.2.1 Pip + CUDA11.8<\/strong><\/h4>\n\n\n\n<div class=\"wp-block-urvanov-syntax-highlighter-code-block\"><pre class=\"lang:sh decode:true \">pip3 install torch torchvision torchaudio --index-url https:\/\/download.pytorch.org\/whl\/cu118<\/pre><\/div>\n\n\n\n<h4 class=\"wp-block-heading\"><strong>6.2.2 Pip + CUDA12.1<\/strong><\/h4>\n\n\n\n<div class=\"wp-block-urvanov-syntax-highlighter-code-block\"><pre class=\"lang:sh decode:true \" >pip3 install torch torchvision torchaudio --index-url https:\/\/download.pytorch.org\/whl\/cu121<\/pre><\/div>\n\n\n\n<h4 class=\"wp-block-heading\"><strong>6.2.3 Pip + CUDA12.4<\/strong><\/h4>\n\n\n\n<div class=\"wp-block-urvanov-syntax-highlighter-code-block\"><pre class=\"lang:python decode:true \" >pip3 install torch torchvision torchaudio<\/pre><\/div>\n\n\n\n<h4 class=\"wp-block-heading\"><strong>6.2.4 Pip + ROCm6.0<\/strong> (AMD)<\/h4>\n\n\n\n<div class=\"wp-block-urvanov-syntax-highlighter-code-block\"><pre class=\"lang:sh decode:true \">pip3 install torch torchvision torchaudio --index-url https:\/\/download.pytorch.org\/whl\/rocm6.0<\/pre><\/div>\n\n\n\n<h3 class=\"wp-block-heading\"><strong>6.3 Mac + Python<\/strong><\/h3>\n\n\n\n<h4 class=\"wp-block-heading\"><strong>6.3.1 Conda<\/strong><\/h4>\n\n\n\n<div class=\"wp-block-urvanov-syntax-highlighter-code-block\"><pre class=\"lang:sh decode:true \">conda install pytorch::pytorch torchvision torchaudio -c pytorch<\/pre><\/div>\n\n\n\n<h4 class=\"wp-block-heading\"><strong>6.3.2 Pip<\/strong><\/h4>\n\n\n\n<div class=\"wp-block-urvanov-syntax-highlighter-code-block\"><pre class=\"lang:sh decode:true \">pip3 install torch torchvision torchaudio<\/pre><\/div>\n\n\n\n<h3 class=\"wp-block-heading\"><strong>6.4 Windows + Python + Conda<\/strong><\/h3>\n\n\n\n<h4 class=\"wp-block-heading\"><strong>6.4.0 Conda + CPU<\/strong><\/h4>\n\n\n\n<div class=\"wp-block-urvanov-syntax-highlighter-code-block\"><pre class=\"lang:sh decode:true \">conda install pytorch torchvision torchaudio cpuonly -c pytorch<\/pre><\/div>\n\n\n\n<h4 class=\"wp-block-heading\"><strong>6.4.1 Conda + CUDA11.8<\/strong><\/h4>\n\n\n\n<div class=\"wp-block-urvanov-syntax-highlighter-code-block\"><pre class=\"lang:sh decode:true \">conda install pytorch torchvision torchaudio pytorch-cuda=11.8 -c pytorch -c nvidia<\/pre><\/div>\n\n\n\n<h4 class=\"wp-block-heading\"><strong>6.4.2 Conda + CUDA12.1<\/strong><\/h4>\n\n\n\n<div class=\"wp-block-urvanov-syntax-highlighter-code-block\"><pre class=\"lang:sh decode:true \">conda install pytorch torchvision torchaudio pytorch-cuda=12.1 -c pytorch -c nvidia<\/pre><\/div>\n\n\n\n<h4 class=\"wp-block-heading\"><strong>6.4.3 Conda + CUDA12.4<\/strong><\/h4>\n\n\n\n<div class=\"wp-block-urvanov-syntax-highlighter-code-block\"><pre class=\"lang:python decode:true \" >conda install pytorch torchvision torchaudio pytorch-cuda=12.4 -c pytorch -c nvidia<\/pre><\/div>\n\n\n\n<p><\/p>\n\n\n\n<h4 class=\"wp-block-heading\"><strong>6.5 Windows + Python + pip<\/strong><\/h4>\n\n\n\n<h4 class=\"wp-block-heading\"><strong>6.5.0 Pip + CPU<\/strong><\/h4>\n\n\n\n<div class=\"wp-block-urvanov-syntax-highlighter-code-block\"><pre class=\"lang:sh decode:true \" >pip3 install torch torchvision torchaudio<\/pre><\/div>\n\n\n\n<h4 class=\"wp-block-heading\"><strong>6.5.1 Pip + CUDA11.8<\/strong><\/h4>\n\n\n\n<div class=\"wp-block-urvanov-syntax-highlighter-code-block\"><pre class=\"lang:sh decode:true \">pip3 install torch torchvision torchaudio --index-url https:\/\/download.pytorch.org\/whl\/cu118<\/pre><\/div>\n\n\n\n<h4 class=\"wp-block-heading\"><strong>6.5.2 Pip + CUDA12.1<\/strong><\/h4>\n\n\n\n<div class=\"wp-block-urvanov-syntax-highlighter-code-block\"><pre class=\"lang:sh decode:true \">pip3 install torch torchvision torchaudio --index-url https:\/\/download.pytorch.org\/whl\/cu121<\/pre><\/div>\n\n\n\n<h4 class=\"wp-block-heading\"><strong>6.5.3 Pip + CUDA12.4<\/strong><\/h4>\n\n\n\n<div class=\"wp-block-urvanov-syntax-highlighter-code-block\"><pre class=\"lang:python decode:true \" >pip3 install torch torchvision torchaudio --index-url https:\/\/download.pytorch.org\/whl\/cu124<\/pre><\/div>\n\n\n\n<h2 class=\"wp-block-heading\"><strong>7. \u5224\u65ad \u5f53\u524d\u4f7f\u7528\u7684 python \u8def\u5f84<\/strong><\/h2>\n\n\n\n<div class=\"wp-block-urvanov-syntax-highlighter-code-block\"><pre class=\"lang:sh decode:true \">python -c \"import sys; print(sys.executable)\"\n<\/pre><\/div>\n\n\n\n<h2 class=\"wp-block-heading\"><strong>8. \u5b89\u88c5 transformers<\/strong><\/h2>\n\n\n\n<h3 class=\"wp-block-heading\"><strong>8.1 \u6b63\u5e38\u5b89\u88c5<\/strong><\/h3>\n\n\n\n<div class=\"wp-block-urvanov-syntax-highlighter-code-block\"><pre class=\"lang:python decode:true \" >pip install transformers<\/pre><\/div>\n\n\n\n<h3 class=\"wp-block-heading\"><strong>8.2 <\/strong>\u66f4\u65b0<\/h3>\n\n\n\n<div class=\"wp-block-urvanov-syntax-highlighter-code-block\"><pre class=\"lang:python decode:true \" >pip install --upgrade transformers\n<\/pre><\/div>\n\n\n\n<h3 class=\"wp-block-heading\"><strong>8.3 <\/strong>\u4f7f\u7528\u6700\u65b0\u7248\u672c(dev)<\/h3>\n\n\n\n<div class=\"wp-block-urvanov-syntax-highlighter-code-block\"><pre class=\"lang:python decode:true \" >pip install git+https:\/\/github.com\/huggingface\/transformers<\/pre><\/div>\n\n\n\n<p><\/p>\n","protected":false},"excerpt":{"rendered":"<p>1. HF \u76f8\u5173\u7684\u73af\u5883\u53d8\u91cf HF\uff08Hugging Face\uff09\u5e93\u4f7f\u7528\u4e86\u591a\u4e2a\u73af\u5883\u53d8\u91cf\u6765\u914d\u7f6e\u5176\u884c\u4e3a\uff0c\u7279\u522b\u662f\u5728\u6a21\u578b\u4e0b\u8f7d [&hellip;]<\/p>\n","protected":false},"author":1,"featured_media":0,"comment_status":"open","ping_status":"open","sticky":false,"template":"","format":"standard","meta":{"site-sidebar-layout":"default","site-content-layout":"","ast-site-content-layout":"default","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":"set","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":""}},"_jetpack_memberships_contains_paid_content":false,"footnotes":""},"categories":[444],"tags":[296],"class_list":["post-3701","post","type-post","status-publish","format-standard","hentry","category-ai","tag-ai"],"views":3232,"jetpack_sharing_enabled":true,"jetpack_featured_media_url":"","_links":{"self":[{"href":"https:\/\/www.aqwu.net\/wp\/index.php?rest_route=\/wp\/v2\/posts\/3701","targetHints":{"allow":["GET"]}}],"collection":[{"href":"https:\/\/www.aqwu.net\/wp\/index.php?rest_route=\/wp\/v2\/posts"}],"about":[{"href":"https:\/\/www.aqwu.net\/wp\/index.php?rest_route=\/wp\/v2\/types\/post"}],"author":[{"embeddable":true,"href":"https:\/\/www.aqwu.net\/wp\/index.php?rest_route=\/wp\/v2\/users\/1"}],"replies":[{"embeddable":true,"href":"https:\/\/www.aqwu.net\/wp\/index.php?rest_route=%2Fwp%2Fv2%2Fcomments&post=3701"}],"version-history":[{"count":42,"href":"https:\/\/www.aqwu.net\/wp\/index.php?rest_route=\/wp\/v2\/posts\/3701\/revisions"}],"predecessor-version":[{"id":4822,"href":"https:\/\/www.aqwu.net\/wp\/index.php?rest_route=\/wp\/v2\/posts\/3701\/revisions\/4822"}],"wp:attachment":[{"href":"https:\/\/www.aqwu.net\/wp\/index.php?rest_route=%2Fwp%2Fv2%2Fmedia&parent=3701"}],"wp:term":[{"taxonomy":"category","embeddable":true,"href":"https:\/\/www.aqwu.net\/wp\/index.php?rest_route=%2Fwp%2Fv2%2Fcategories&post=3701"},{"taxonomy":"post_tag","embeddable":true,"href":"https:\/\/www.aqwu.net\/wp\/index.php?rest_route=%2Fwp%2Fv2%2Ftags&post=3701"}],"curies":[{"name":"wp","href":"https:\/\/api.w.org\/{rel}","templated":true}]}}