{"id":2491,"date":"2023-02-16T15:35:36","date_gmt":"2023-02-16T07:35:36","guid":{"rendered":"http:\/\/nick.txtcc.com\/?p=2491"},"modified":"2023-09-04T15:47:16","modified_gmt":"2023-09-04T07:47:16","slug":"windows10%e5%ae%89%e8%a3%85tensorflow-gpu%e8%bf%87%e7%a8%8b%ef%bc%88gtx-3060%ef%bc%89","status":"publish","type":"post","link":"https:\/\/nick.txtcc.com\/index.php\/win\/2491","title":{"rendered":"Windows10\u5b89\u88c5Tensorflow-GPU\u8fc7\u7a0b\uff08GTX 3060\uff09"},"content":{"rendered":"

\u7b2c\u4e00\u6b65, \u786e\u8ba4Tensorflow\u7248\u672c\u5bf9\u5e94\u7684\u4e09\u4e2a\u7248\u672c\u4fe1\u606f: \u9a71\u52a8\u7248\u672c\u53f7, cuda\u7248\u672c\u53f7, cudnn\u7248\u672c\u53f7<\/p>\n

\u5230\u4e0b\u9762\u7f51\u5740\u67e5\u770btf\u7248\u672c\u5bf9\u5e94\u7684\u4e24\u4e2a\u7248\u672c\u4fe1\u606f\nhttps:\/\/www.tensorflow.org\/install\/source_windows?hl=zh-cn#gpu<\/a><\/p>\n

\u6211\u7528tensorflow_gpu-2.6.0, \u5bf9\u5e94\u7684\u5c31\u662fCUDA 11.2, cuDNN 8.1\n https:\/\/developer.nvidia.com\/cuda-toolkit-archive<\/a><\/a>\n https:\/\/developer.nvidia.com\/rdp\/cudnn-archive<\/a><\/a><\/p>\n

~~\u4e24\u4e2a\u5148\u4e0b\u8f7d\u597d, \u7136\u540e\u627eCUDA 11.2\u5bf9\u5e94\u7684\u9a71\u52a8\u5b89\u88c5, \u4e0b\u9762\u94fe\u63a5\u7b2c2\u4e2atable<\/del>\n~~https:\/\/docs.nvidia.com\/cuda\/cuda-toolkit-release-notes\/index.html<\/a><\/a><\/del><\/p>\n

CUDA 11.2.x\u00a0 \u5bf9\u5e94\u7684\u7248\u672c\u53f7\u533a\u95f4\u662f >=452.39, \u5230\u4e0b\u9762\u7f51\u5740\u641c\u7d22\u5bf9\u5e94\u663e\u5361\u548c\u5bf9\u5e94\u7248\u672c\u7684\u9a71\u52a8<\/del>\n https:\/\/www.nvidia.cn\/Download\/Find.aspx?lang=cn<\/a><\/a><\/del><\/p>\n

~~\u6211\u4e0b\u8f7d\u7684\u662fGeForce Game Ready 472.12\u7248\u672c\u9a71\u52a8, \u5728\u9a71\u52a8\u4fe1\u606f\u91cc\u770b\u5230 NVCUDA64.DLL \u7248\u672c\u4fe1\u606f\u662f 11.4.136<\/del><\/p>\n~~

wsl \u7248\u672c\u5b89\u88c5\uff1a https:\/\/developer.nvidia.cn\/cuda-downloads?target_os=Linux&target_arch=x86_64&Distribution=WSL-Ubuntu&target_version=2.0&target_type=deb_local<\/a><\/p>\n

<span class="cudaBash">wget https:\/\/developer.download.nvidia.com\/compute\/cuda\/repos\/wsl-ubuntu\/x86_64\/cuda-wsl-ubuntu.pin\n<\/span>\n<span class="cudaBash">sudo mv cuda-wsl-ubuntu.pin \/etc\/apt\/preferences.d\/cuda-repository-pin-600\n<\/span>\n<span class="cudaBash">wget https:\/\/developer.download.nvidia.com\/compute\/cuda\/12.2.2\/local_installers\/cuda-repo-wsl-ubuntu-12-2-local_12.2.2-1_amd64.deb\n<\/span>\n<span class="cudaBash">sudo dpkg -i cuda-repo-wsl-ubuntu-12-2-local_12.2.2-1_amd64.deb\n<\/span>\n<span class="cudaBash">sudo cp \/var\/cuda-repo-wsl-ubuntu-12-2-local\/cuda-*-keyring.gpg \/usr\/share\/keyrings\/\n<\/span>\n<span class="cudaBash">sudo apt-get update\n<\/span>\n<span class="cudaBash">sudo apt-get -y install cuda<\/span><\/code><\/pre>\n <\/p>\n
\u5b89\u88c5\u597d\u9a71\u52a8, cuda, cudnn\u540e, \u8bbe\u7f6e\u7cfb\u7edf\u73af\u5883\u53d8\u91cf, \u5728PATH\u91cc\u6dfb\u52a0\u4e24\u4e2a:\n%CUDA_PATH%\\bin\n%CUDA_PATH%\\lib\\x64<\/p>\n
\u68c0\u6d4b\u662f\u5426\u5b89\u88c5\u6210\u529f<\/p>\n
cmd\u547d\u4ee4\u884c\u8f93\u5165\uff1anvidia-smi<\/p>\n
 <\/p>\n
\u5728\u7cfb\u7edf\u53d8\u91cf\u4e2d\uff0c\u589e\u52a0\u53d8\u91cf\u540dTF_GEFORCE_GPU_ALLOW_GROWTH\uff0c\u53d6\u503c\u4e3atrue<\/p>\n
 <\/p>\n
\u63a5\u4e0b\u6765\u662f\u5b89\u88c5tensorflow_gpu<\/p>\n\u5728conda\u91cc\u521b\u5efapython 3.7\u73af\u5883, \u5728\u73af\u5883\u91cc\u5b89\u88c5tf, \u6211\u662f\u624b\u52a8\u4e0b\u8f7d\u7684\u5b89\u88c5\u5305, \u4e0b\u8f7d\u5bf9\u5e94py3.7\u7248\u672c\u7684:\nhttps:\/\/www.tensorflow.org\/install\/pip.html?hl=zh-cn#package-location<\/a><\/p>\n
\u7528pip\u5b89\u88c5\u4e0b\u8f7d\u7684tf\u5b89\u88c5\u5305\u5373\u53ef<\/p>\n
 <\/p>\n
\u6d4b\u8bd5\u4ee3\u7801\u5982\u4e0b:\nimport tensorflow as tf\nprint(‘GPU’,tf.config.list_physical_devices(‘GPU’))\nprint(“Num GPUs Available: “, len(tf.config.experimental.list_physical_devices(‘GPU’)))<\/p>\n
\u6211\u53ea\u6709\u4e00\u5757GTX3060\u663e\u5361\uff0c\u6267\u884c\u540e\u8fd4\u56de:<\/p>\n
GPU [PhysicalDevice(name=’\/physical_device:GPU:0′, device_type=’GPU’)]\nNum GPUs Available: 1\n\u5230\u6b64\u4e3a\u5b89\u88c5\u6210\u529f\u3002<\/p>\n
 <\/p>\n
 <\/p>\n
\u5b89\u88c5pytorch\u5bf9\u5e94cuda 11.1\u7248\u672c, \u5230\u4ee5\u4e0b\u94fe\u63a5\u4e0b\u8f7d\u5b89\u88c5\u5305:\nhttps:\/\/download.pytorch.org\/whl\/torch_stable.html<\/a><\/p>\n
\u5206\u522b\u662f:\ntorch-1.9.0+cu111-cp37-cp37m-win_amd64.whl\ntorchvision-0.10.0+cpu-cp37-cp37m-win_amd64.whl\ntorchaudio-0.9.0-cp37-cp37m-win_amd64.whl<\/p>\n
\u7528pip\u5b89\u88c5\u4ee5\u4e0a3\u4e2a\u5b89\u88c5\u5305\u5373\u53ef<\/p>\n
\u6d4b\u8bd5\u4ee3\u7801:<\/p>\n
import torch<\/p>\n
\u8fd4\u56de\u5f53\u524d\u8bbe\u5907\u7d22\u5f15<\/h1>\nprint(torch.cuda.current_device())<\/p>\n
\u8fd4\u56deGPU\u7684\u6570\u91cf<\/h1>\nprint(torch.cuda.device_count())<\/p>\n
\u8fd4\u56degpu\u540d\u5b57\uff0c\u8bbe\u5907\u7d22\u5f15\u9ed8\u8ba4\u4ece0\u5f00\u59cb<\/h1>\nprint(torch.cuda.get_device_name(0))<\/p>\ncuda\u662f\u5426\u53ef\u7528\uff0c\u4e3atrue\u5c31\u53ef\u7528\u4e86<\/h1>\nprint(torch.cuda.is_available())\n#\u67e5\u770bcudnn\u7248\u672c\nprint(torch.backends.cudnn.version())\n#\u67e5\u770btorch\u7248\u672c\nprint(torch.version<\/strong>)<\/p>\n
 <\/p>","protected":false},"excerpt":{"rendered":"\u7b2c\u4e00\u6b65, \u786e\u8ba4Tensorflow\u7248\u672c\u5bf9\u5e94\u7684\u4e09\u4e2a\u7248\u672c\u4fe1\u606f: \u9a71\u52a8\u7248\u672c\u53f7, cuda\u7248\u672c\u53f7, cudnn\u7248\u672c\u53f7 \u5230\u4e0b\u9762\u7f51\u5740\u67e5\u770btf\u7248\u672c\u5bf9\u5e94\u7684\u4e24\u4e2a\u7248\u672c\u4fe1\u606f https:\/\/www.tensorflow.o..&#46<\/a>;<\/p>","protected":false},"author":1,"featured_media":0,"comment_status":"closed","ping_status":"closed","sticky":false,"template":"","format":"standard","meta":{"footnotes":""},"categories":[3],"tags":[],"class_list":["post-2491","post","type-post","status-publish","format-standard","hentry","category-win"],"_links":{"self":[{"href":"https:\/\/nick.txtcc.com\/index.php\/wp-json\/wp\/v2\/posts\/2491","targetHints":{"allow":["GET"]}}],"collection":[{"href":"https:\/\/nick.txtcc.com\/index.php\/wp-json\/wp\/v2\/posts"}],"about":[{"href":"https:\/\/nick.txtcc.com\/index.php\/wp-json\/wp\/v2\/types\/post"}],"author":[{"embeddable":true,"href":"https:\/\/nick.txtcc.com\/index.php\/wp-json\/wp\/v2\/users\/1"}],"replies":[{"embeddable":true,"href":"https:\/\/nick.txtcc.com\/index.php\/wp-json\/wp\/v2\/comments?post=2491"}],"version-history":[{"count":6,"href":"https:\/\/nick.txtcc.com\/index.php\/wp-json\/wp\/v2\/posts\/2491\/revisions"}],"predecessor-version":[{"id":2592,"href":"https:\/\/nick.txtcc.com\/index.php\/wp-json\/wp\/v2\/posts\/2491\/revisions\/2592"}],"wp:attachment":[{"href":"https:\/\/nick.txtcc.com\/index.php\/wp-json\/wp\/v2\/media?parent=2491"}],"wp:term":[{"taxonomy":"category","embeddable":true,"href":"https:\/\/nick.txtcc.com\/index.php\/wp-json\/wp\/v2\/categories?post=2491"},{"taxonomy":"post_tag","embeddable":true,"href":"https:\/\/nick.txtcc.com\/index.php\/wp-json\/wp\/v2\/tags?post=2491"}],"curies":[{"name":"wp","href":"https:\/\/api.w.org\/{rel}","templated":true}]}}

\u8fd4\u56deGPU\u7684\u6570\u91cf<\/h1>\n
print(torch.cuda.device_count())<\/p>\n

\u8fd4\u56degpu\u540d\u5b57\uff0c\u8bbe\u5907\u7d22\u5f15\u9ed8\u8ba4\u4ece0\u5f00\u59cb<\/h1>\n
print(torch.cuda.get_device_name(0))<\/p>\n