{"id":1110,"date":"2024-10-26T02:30:30","date_gmt":"2024-10-26T02:30:30","guid":{"rendered":"https:\/\/blue16.cn\/?p=1110"},"modified":"2024-10-26T02:30:32","modified_gmt":"2024-10-26T02:30:32","slug":"cuda%e4%b8%8epytorch%e5%ae%89%e8%a3%85%e6%95%99%e7%a8%8b","status":"publish","type":"post","link":"https:\/\/blue16.cn\/index.php\/2024\/10\/26\/1110\/","title":{"rendered":"CUDA\u4e0ePytorch\u5b89\u88c5\u6559\u7a0b"},"content":{"rendered":"\n<h2 class=\"wp-block-heading\">\u524d\u8a00<\/h2>\n\n\n\n<p>\u641e\u6df1\u5ea6\u5b66\u4e60\u7684\uff0cCUDA\u80af\u5b9a\u5fc5\u4e0d\u53ef\u5c11\uff0c\u53ea\u4e0d\u8fc7\u636e\u6211\u6240\u77e5\uff0c\u7b2c\u4e00\u6b21\u5b89\u88c5CUDA\u662f\u975e\u5e38\u75db\u82e6\u7684\uff0c\u66f4\u522b\u8bf4Pytorch\u4e86\uff0c\u4e0d\u7ba1\u662f\u4e0b\u8f7d\u8fd8\u662f\u62a5\u9519\u3002<\/p>\n\n\n\n<h2 class=\"wp-block-heading\">\u9700\u8981\u51c6\u5907\u7684<\/h2>\n\n\n\n<ul class=\"wp-block-list\">\n<li>Anaconda\uff08\u63a8\u8350\uff09\/MiniConda<\/li>\n\n\n\n<li>\u4f18\u5316\u7f51\u7edc\u73af\u5883<\/li>\n<\/ul>\n\n\n\n<h2 class=\"wp-block-heading\">\u64cd\u4f5c\u65b9\u5f0f<\/h2>\n\n\n\n<h3 class=\"wp-block-heading\">1.\u4e0b\u8f7d\u9a71\u52a8\/\u66f4\u65b0\u663e\u5361\u9a71\u52a8\uff08\u53ef\u4ee5\u8df3\u8fc7\u4f46\u63a8\u8350\u66f4\u65b0\u4e00\u4e0b\uff0c\u53ef\u4ee5\u63d0\u5347CUDA\u652f\u6301\u7248\u672c\uff09<\/h3>\n\n\n\n<p><a href=\"https:\/\/www.nvidia.cn\/geforce\/drivers\/\">NVIDIA GeForce \u9a71\u52a8\u7a0b\u5e8f &#8211; N \u5361\u9a71\u52a8 | NVIDIA<\/a><\/p>\n\n\n\n<p>\u5982\u679c\u8981\u6253\u6e38\u620f\u7684\uff0c\u4e0b\u8f7dGameReady\uff0c\u5982\u679c\u4e0d\u6253\u6e38\u620f\u6216\u8005\u6253\u7684\u5c11\uff0c\u7ecf\u5e38\u505a\u89c6\u9891\u526a\u8f91\u56fe\u50cf\u5904\u7406\u7684\uff0c\u4e0b\u8f7dStudio\u3002\u8fd9\u91cc\u4e0b\u8f7d\u5b8c\u5b89\u88c5\u5305\u540e\uff0c\u52fe\u9009\u201c\u81ea\u5b9a\u4e49\u201d\uff0c\u7136\u540e\u63a8\u8350\u6267\u884c\u201c\u6e05\u6d01\u5b89\u88c5\u201d\uff0c\u4e5f\u5c31\u662f\u5378\u8f7d\u539f\u7248\u9a71\u52a8\u540e\u518d\u5b89\u88c5\u3002<\/p>\n\n\n\n<h3 class=\"wp-block-heading\">2.\u67e5\u8be2\u663e\u5361\u6700\u9ad8\u652f\u6301\u7684CUDA\u7248\u672c<\/h3>\n\n\n\n<p>\u6253\u5f00cmd\uff0c\u6267\u884c\uff1anvidia-smi<\/p>\n\n\n\n<figure class=\"wp-block-image size-large\"><div class='fancybox-wrapper lazyload-container-unload' data-fancybox='post-images' href='https:\/\/blue16.cn\/wp-content\/uploads\/2024\/10\/image-50-1024x467.png'><img class=\"lazyload lazyload-style-1\" src=\"data:image\/svg+xml;base64,PCEtLUFyZ29uTG9hZGluZy0tPgo8c3ZnIHdpZHRoPSIxIiBoZWlnaHQ9IjEiIHhtbG5zPSJodHRwOi8vd3d3LnczLm9yZy8yMDAwL3N2ZyIgc3Ryb2tlPSIjZmZmZmZmMDAiPjxnPjwvZz4KPC9zdmc+\"  loading=\"lazy\" decoding=\"async\" width=\"1024\" height=\"467\" data-original=\"https:\/\/blue16.cn\/wp-content\/uploads\/2024\/10\/image-50-1024x467.png\" src=\"data:image\/png;base64,iVBORw0KGgoAAAANSUhEUgAAAAEAAAABCAYAAAAfFcSJAAAAAXNSR0IArs4c6QAAAARnQU1BAACxjwv8YQUAAAAJcEhZcwAADsQAAA7EAZUrDhsAAAANSURBVBhXYzh8+PB\/AAffA0nNPuCLAAAAAElFTkSuQmCC\" alt=\"\" class=\"wp-image-1111\"  sizes=\"auto, (max-width: 1024px) 100vw, 1024px\" \/><\/div><\/figure>\n\n\n\n<p>\u8fd9\u91cc\u7684\u201cCUDA Version\u201d\u6307\u7684\u662f\u4f60\u7684\u663e\u5361\u6700\u9ad8\u652f\u6301\u7248\u672c\uff0c\u4e5f\u5c31\u662f\u4f60\u5b89\u88c5\u7684CUDA\u7248\u672c\u4e0d\u80fd\u9ad8\u4e8e\u8fd9\u4e2a\u7248\u672c\u53f7\uff0c\u4f46\u662f\u53ef\u4ee5\u66f4\u4f4e\u3002<\/p>\n\n\n\n<h3 class=\"wp-block-heading\">3.\u67e5\u8be2PyTorch\u6700\u9ad8\u652f\u6301\u7248\u672c\u53f7<\/h3>\n\n\n\n<p>\u8fdb\u5165<a href=\"https:\/\/pytorch.org\/get-started\/locally\/\">Start Locally | PyTorch<\/a>\uff0c\u627e\u5230\u4e0b\u9762\u7684\u90e8\u5206\uff1a<\/p>\n\n\n\n<figure class=\"wp-block-image size-large\"><div class='fancybox-wrapper lazyload-container-unload' data-fancybox='post-images' href='https:\/\/blue16.cn\/wp-content\/uploads\/2024\/10\/image-52-1024x470.png'><img class=\"lazyload lazyload-style-1\" src=\"data:image\/svg+xml;base64,PCEtLUFyZ29uTG9hZGluZy0tPgo8c3ZnIHdpZHRoPSIxIiBoZWlnaHQ9IjEiIHhtbG5zPSJodHRwOi8vd3d3LnczLm9yZy8yMDAwL3N2ZyIgc3Ryb2tlPSIjZmZmZmZmMDAiPjxnPjwvZz4KPC9zdmc+\"  loading=\"lazy\" decoding=\"async\" width=\"1024\" height=\"470\" data-original=\"https:\/\/blue16.cn\/wp-content\/uploads\/2024\/10\/image-52-1024x470.png\" src=\"data:image\/png;base64,iVBORw0KGgoAAAANSUhEUgAAAAEAAAABCAYAAAAfFcSJAAAAAXNSR0IArs4c6QAAAARnQU1BAACxjwv8YQUAAAAJcEhZcwAADsQAAA7EAZUrDhsAAAANSURBVBhXYzh8+PB\/AAffA0nNPuCLAAAAAElFTkSuQmCC\" alt=\"\" class=\"wp-image-1113\"  sizes=\"auto, (max-width: 1024px) 100vw, 1024px\" \/><\/div><\/figure>\n\n\n\n<p>\u6211\u4eec\u9009\u62e9Windows\u3001Conda\u3001Python\uff0c\u8fd9\u91cc\u7684\u201cCompute Platform\u201d\u5c31\u662f\u5f53\u524dPytorch\u652f\u6301\u7684\u6700\u9ad8\u7248\u672c\uff0c\u5f53\u524d\u662f12.4\uff0c\u6211\u4eec\u663e\u5361\u6700\u9ad8\u652f\u6301\u662f12.7\uff0c\u662f\u652f\u6301\u7684\u3002\u8fd9\u4e5f\u662f\u6211\u4eec\u9700\u8981\u4e0b\u8f7d\u7684CUDA Toolkit\u7684\u7248\u672c\u3002<\/p>\n\n\n\n<h3 class=\"wp-block-heading\">4.\u8fdb\u5165NVIDIA CUDA\u5b98\u7f51\u4e0b\u8f7d\u5bf9\u5e94\u7248\u672c\u7684CUDA<\/h3>\n\n\n\n<p><a href=\"https:\/\/developer.nvidia.com\/cuda-toolkit-archive\">CUDA Toolkit Archive | NVIDIA Developer<\/a><\/p>\n\n\n\n<p>\u6ce8\u610f\u662f\u201cToolkit Archive\u201d\uff0c\u56e0\u4e3a\u6211\u4eec<strong>\u4e0d\u80fd\u8fdb\u5165\u5b98\u7f51\u4e0b\u8f7d\u6700\u65b0\u7248\u672c\u7684CUDA<\/strong>\uff0c\u6700\u65b0\u7248\u4f1a\u5bfc\u81f4PyTorch\u4e0d\u8bc6\u522b\u5bfc\u81f4\u62a5\u9519\u3002\u627e\u5230\u5bf9\u5e94\u7248\u672c\uff0c\u76ee\u524d\u6211\u8fd9\u91cc\u662f12.4\uff0c\u56e0\u6b64\u4e0b\u8f7d12.4.0\u7684\u7248\u672c\uff1a<\/p>\n\n\n\n<figure class=\"wp-block-image size-large\"><div class='fancybox-wrapper lazyload-container-unload' data-fancybox='post-images' href='https:\/\/blue16.cn\/wp-content\/uploads\/2024\/10\/image-53-1024x388.png'><img class=\"lazyload lazyload-style-1\" src=\"data:image\/svg+xml;base64,PCEtLUFyZ29uTG9hZGluZy0tPgo8c3ZnIHdpZHRoPSIxIiBoZWlnaHQ9IjEiIHhtbG5zPSJodHRwOi8vd3d3LnczLm9yZy8yMDAwL3N2ZyIgc3Ryb2tlPSIjZmZmZmZmMDAiPjxnPjwvZz4KPC9zdmc+\"  loading=\"lazy\" decoding=\"async\" width=\"1024\" height=\"388\" data-original=\"https:\/\/blue16.cn\/wp-content\/uploads\/2024\/10\/image-53-1024x388.png\" src=\"data:image\/png;base64,iVBORw0KGgoAAAANSUhEUgAAAAEAAAABCAYAAAAfFcSJAAAAAXNSR0IArs4c6QAAAARnQU1BAACxjwv8YQUAAAAJcEhZcwAADsQAAA7EAZUrDhsAAAANSURBVBhXYzh8+PB\/AAffA0nNPuCLAAAAAElFTkSuQmCC\" alt=\"\" class=\"wp-image-1114\"  sizes=\"auto, (max-width: 1024px) 100vw, 1024px\" \/><\/div><\/figure>\n\n\n\n<p>\u4e0b\u8f7d\u540e\u548c\u5b89\u88c5\u663e\u5361\u9a71\u52a8\u4e00\u6837\u5b89\u88c5\u5373\u53ef\u3002<\/p>\n\n\n\n<p>Tips\uff1a\u8fd9\u4e2a\u7f51\u7ad9\u5373\u4f7f\u662f\u4f18\u5316\u4e86\u7f51\u7edc\u8fd8\u662f\u4f1a\u5361\uff0c\u662f\u6b63\u5e38\u7684\uff0c\u53ef\u80fd\u6bd4\u8f83\u571f\u8c46\uff0c\u9700\u8981\u591a\u6b21\u5c1d\u8bd5\u3002\u3002\u3002\u597d\u50cf\u6700\u8fd1Nvidia\u4fee\u590d\u4e86\uff0c\u53cd\u6b63\u53bb\u5e74\u90fd\u662f\u975e\u5e38\u5361\u7684\u3002<\/p>\n\n\n\n<h3 class=\"wp-block-heading\">5.\u521b\u5efa\u4e00\u4e2aAnaconda\u73af\u5883\uff0c\u7136\u540e\u5b89\u88c5PyTorch<\/h3>\n\n\n\n<p>\u9996\u5148\u521b\u5efa\u5e76\u6fc0\u6d3b\u4e00\u4e2aAnaconda\u73af\u5883\uff1a<\/p>\n\n\n\n<pre class=\"wp-block-code\"><code>conda create -n pytorch_env\nconda activate pytorch_env<\/code><\/pre>\n\n\n\n<p>\u7136\u540e\u590d\u5236\u521a\u624dPyTorch\u5b98\u7f51\u4e0a\u7684\u5bf9\u5e94\u6307\u4ee4\uff0c\u6267\u884c\u4e00\u4e0b\uff0c\u4f8b\u5982\u6211\u8fd9\u91cc\u662f\uff1a<\/p>\n\n\n\n<pre class=\"wp-block-code\"><code>conda install pytorch torchvision torchaudio pytorch-cuda=12.4 -c pytorch -c nvidia<\/code><\/pre>\n\n\n\n<p>\u6ce8\u610f<strong>\u5168\u7a0b\u4f18\u5316\u7f51\u7edc\u73af\u5883<\/strong>\uff0c\u4e0d\u4f1a\u4f18\u5316\u7684\u95ee\u4e0b\u4f60\u4eec\u7684\u5b66\u957f\u5b66\u59d0\uff0c\u6211\u4e2a\u4eba\u559c\u6b22\u6302\u4ee3\u7406\uff0c\u56e0\u4e3a\u6362\u6e90\u9ebb\u70e6\u800c\u4e14\u5bb9\u6613\u51fa\u4e00\u4e9b\u83ab\u540d\u5176\u5999\u7684\u9519\u8bef\u3002<\/p>\n\n\n\n<h3 class=\"wp-block-heading\">6.\u9a8c\u8bc1PyTorch\u662f\u5426\u5b89\u88c5\u6210\u529f<\/h3>\n\n\n\n<p>\u7b49\u8fdb\u5ea6\u6761\u8d70\u5b8c\uff0c\u6267\u884c\u4e0b\u9762\u7684\u4ee3\u7801\uff0c\u6839\u636e\u68c0\u6d4b\u7ed3\u679c\u5224\u65ad\u5373\u53ef\u3002<\/p>\n\n\n\n<pre class=\"wp-block-code\"><code>import torch\n\n# \u6253\u5370PyTorch\u7248\u672c\nprint(f\"PyTorch version: {torch.__version__}\")\n\n# \u68c0\u67e5\u662f\u5426\u6709CUDA\u652f\u6301\nif torch.cuda.is_available():\n    print(\"CUDA is available!\")\n    print(f\"CUDA version: {torch.version.cuda}\")\n    print(f\"Device name: {torch.cuda.get_device_name(0)}\")\n\n    # \u521b\u5efa\u4e00\u4e2a\u5728GPU\u4e0a\u7684\u5f20\u91cf\n    x = torch.rand(5).cuda()\n    print(f\"Tensor on GPU: {x}\")\nelse:\n    print(\"CUDA is not available.\")<\/code><\/pre>\n\n\n\n<p>\u6838\u5fc3\u5c31\u662f\u4e00\u4e2a\u547d\u4ee4\uff1a\u201ctorch.cuda.is_available()\u201d\u3002<\/p>\n\n\n\n<h2 class=\"wp-block-heading\">\u4e00\u4e9b\u5e38\u89c1\u7684\u95ee\u9898<\/h2>\n\n\n\n<h3 class=\"wp-block-heading\">\u76f2\u76ee\u5b89\u88c5CUDA\u7248\u672c\u5bfc\u81f4\u4e0ePyTorch\u4e0d\u5339\u914d<\/h3>\n\n\n\n<p>CUDA\u4e0d\u662f\u663e\u5361\u6700\u9ad8\u652f\u6301\u591a\u5c11\u5c31\u5b89\u88c5\u591a\u9ad8\u7248\u672c\u7684CUDA\uff0cCUDA\u4e0d\u662f\u5411\u4e0b\u517c\u5bb9\u7684\uff0c\u9700\u8981\u5230PyTorch\u5b98\u7f51\u770b\u4e0b\u6700\u9ad8\u652f\u6301\u7248\u672c\u662f\u591a\u5c11\uff0c\u518d\u5b89\u88c5\u5bf9\u5e94\u7684CUDA\u7248\u672c\u3002<\/p>\n\n\n\n<h3 class=\"wp-block-heading\">\u5b89\u88c5\u7684PyTorch\u53ea\u8c03\u7528CPU\u8bad\u7ec3<\/h3>\n\n\n\n<p>\u5b89\u88c5\u6210CPU\u7248\u672c\u7684PyTorch\u4e86\uff0c\u4e0d\u80fd\u7b80\u5355\u4f7f\u7528\u201cpip install pytorch\u201d\u7b49\u6307\u4ee4\u8fdb\u884c\u5b89\u88c5\uff0c\u8bf7\u5378\u8f7d\u540e\uff08\u5efa\u8bae\u76f4\u63a5\u5220\u9664anaconda\u5bf9\u5e94\u73af\u5883\uff09\u518d\u4f7f\u7528PyTorch\u5b98\u7f51\u4e0a\u7684\u6307\u4ee4\u8fdb\u884c\u91cd\u65b0\u5b89\u88c5\u3002<\/p>\n","protected":false},"excerpt":{"rendered":"<p>\u524d\u8a00 \u641e\u6df1\u5ea6\u5b66\u4e60\u7684\uff0cCUDA\u80af\u5b9a\u5fc5\u4e0d\u53ef\u5c11\uff0c\u53ea\u4e0d\u8fc7\u636e\u6211\u6240\u77e5\uff0c\u7b2c\u4e00\u6b21\u5b89\u88c5CUDA\u662f\u975e\u5e38\u75db\u82e6\u7684\uff0c\u66f4\u522b\u8bf4Pytorch [&hellip;]<\/p>\n","protected":false},"author":1,"featured_media":0,"comment_status":"open","ping_status":"open","sticky":false,"template":"","format":"standard","meta":{"footnotes":""},"categories":[41,64,26,27],"tags":[],"class_list":["post-1110","post","type-post","status-publish","format-standard","hentry","category-python","category-pytorch","category-study","category-study_note"],"_links":{"self":[{"href":"https:\/\/blue16.cn\/index.php\/wp-json\/wp\/v2\/posts\/1110","targetHints":{"allow":["GET"]}}],"collection":[{"href":"https:\/\/blue16.cn\/index.php\/wp-json\/wp\/v2\/posts"}],"about":[{"href":"https:\/\/blue16.cn\/index.php\/wp-json\/wp\/v2\/types\/post"}],"author":[{"embeddable":true,"href":"https:\/\/blue16.cn\/index.php\/wp-json\/wp\/v2\/users\/1"}],"replies":[{"embeddable":true,"href":"https:\/\/blue16.cn\/index.php\/wp-json\/wp\/v2\/comments?post=1110"}],"version-history":[{"count":1,"href":"https:\/\/blue16.cn\/index.php\/wp-json\/wp\/v2\/posts\/1110\/revisions"}],"predecessor-version":[{"id":1115,"href":"https:\/\/blue16.cn\/index.php\/wp-json\/wp\/v2\/posts\/1110\/revisions\/1115"}],"wp:attachment":[{"href":"https:\/\/blue16.cn\/index.php\/wp-json\/wp\/v2\/media?parent=1110"}],"wp:term":[{"taxonomy":"category","embeddable":true,"href":"https:\/\/blue16.cn\/index.php\/wp-json\/wp\/v2\/categories?post=1110"},{"taxonomy":"post_tag","embeddable":true,"href":"https:\/\/blue16.cn\/index.php\/wp-json\/wp\/v2\/tags?post=1110"}],"curies":[{"name":"wp","href":"https:\/\/api.w.org\/{rel}","templated":true}]}}