{"id":8246,"date":"2024-02-02T00:20:01","date_gmt":"2024-02-01T16:20:01","guid":{"rendered":"https:\/\/yanjingang.com\/blog\/?p=8246"},"modified":"2026-06-18T15:43:41","modified_gmt":"2026-06-18T07:43:41","slug":"%e5%8d%8e%e7%a1%95rog%e5%ae%89%e8%a3%85ubuntu20-04cuda11-8cudnn8-7","status":"publish","type":"post","link":"https:\/\/yanjingang.com\/blog\/?p=8246","title":{"rendered":"Ubuntu20.04\u90e8\u7f72CUDA11.8\/cuDNN8.7"},"content":{"rendered":"<p>\u672c\u6587\u4e3b\u8981\u8bb0\u5f55\u534e\u7855\u7684GPU\u672c\uff08RTX2070\uff09\u5982\u4f55Ubuntu20.04 + Nvidia\u663e\u5361\u9a71\u52a8 + CUDA11.8 + cuDNN8.7\uff0c\u5e76\u9a8c\u8bc1GPU CUDA\u751f\u6548\u7684\u8fc7\u7a0b\u3002<\/p>\n<h1>\u4e00\u3001\u5b89\u88c5Ubuntu20.04<\/h1>\n<h3><span id=\"1ubuntu184iso\">1.\u4e0b\u8f7dUbuntu20.4\u7684iso\u955c\u50cf\u6587\u4ef6<\/span><\/h3>\n<p>\u81ea\u5df1\u53bb\u5b98\u7f51\u4e0b\u8f7d\uff1ahttps:\/\/www.ubuntu.com\/download\/desktop<\/p>\n<h3><span id=\"2isou\">2.\u628aISO\u955c\u50cf\u70e7\u5f55\u5230U\u76d8\u91cc<\/span><\/h3>\n<p>\u63a8\u8350\u4f7f\u7528<a href=\"https:\/\/www.balena.io\/etcher\/\">balenaEtcher<\/a>\uff0c\u5b83\u53ef\u4ee5\u5728 Linux\u3001Windows \u548c Mac OS \u4e0a\u8fd0\u884c\u3002<\/p>\n<h3><span id=\"3ubuntu\">3.\u51c6\u5907Ubuntu\u78c1\u76d8<\/span><\/h3>\n<p>\u78c1\u76d8\u7ba1\u7406\uff0c\u9009\u4e2dc\u76d8\uff08\u9ed8\u8ba4\u53ea\u6709\u4e00\u4e2ac\u76d8\uff09\uff0c\u53f3\u952e\u538b\u7f29\u5377\uff0c\u586b\u5199\u8981\u4ecec\u76d8\u4e2d\u5206\u5272\u51fa\u6765\u7684\u78c1\u76d8\u5927\u5c0f\uff0c\u8fd9\u91cc\u586b\u7684512000M\u5373500G\u7ed9ubuntu\u3002\u6ce8\u610f\u4e0d\u8981\u683c\u5f0f\u5316<\/p>\n<h3><span id=\"4F2bois\">4.\u91cd\u542f\u6309F2\u4fee\u6539Bois\u8bbe\u7f6e<\/span><\/h3>\n<h6><span id=\"41_Secure_Boot_Disable\">4.1 Secure Boot \u8bbe\u7f6e\u6210Disable<\/span><\/h6>\n<p>\u5426\u5219\u4f1a\u9ed1\u5c4f<\/p>\n<h6><span id=\"42_BootUwin10\">4.2 Boot\u542f\u52a8\u987a\u5e8f\u6539\u4e3aU\u76d8\u5728win10\u524d\u8fb9<\/span><\/h6>\n<h6><span id=\"43_STAT_ModelAHCI\">4.3 STAT Model\u6539\u4e3aAHCI<\/span><\/h6>\n<p>\u5426\u5219ubuntu\u5206\u533a\u65f6\u4f1a\u627e\u4e0d\u5230\u78c1\u76d8\uff1b\u4f46\u662f\u6539\u4e3aAHCI\u4f1a\u5bfc\u81f4\u8fdb\u4e0d\u4e86win10\uff0c\u6682\u65f6\u6ca1\u627e\u5230\u66f4\u597d\u7684\u529e\u6cd5<\/p>\n<h3><span id=\"5ubuntu\">5.\u5b89\u88c5Ubuntu<\/span><\/h3>\n<p>\u5982\u679c\u9047\u5230No EFI System Partition was found.this system will likely\u2026.. Go back \u2026.\u65f6\uff0c\u9700\u8981\u65b0\u5efa\u4e00\u4e2a\u7528\u4e8eEFI\u7684\u5206\u533a\uff0c\u7ed9100M\u5c31\u53ef\u4ee5\u3002<\/p>\n<h1><span id=\"6ubuntugpu\">\u4e8c\u3001\u5b89\u88c5GPU\u9a71\u52a8<\/span><\/h1>\n<h3><span id=\"61\">1 \u51c6\u5907\u5de5\u4f5c<\/span><\/h3>\n<pre class=\"pure-highlightjs\"><code class=\"hljs sql\">sudo apt-get <span class=\"hljs-keyword\">install<\/span> vim-gtk openssh-<span class=\"hljs-keyword\">server linux-headers-$(uname -r)<\/span>\r\n\r\nsudo vim \/etc\/<span class=\"hljs-keyword\">default<\/span>\/grub\r\n  GRUB_GFXMODE=<span class=\"hljs-number\">1920<\/span>x1080\r\n\r\nsudo <span class=\"hljs-keyword\">update<\/span>-grub\r\nsudo apt-<span class=\"hljs-keyword\">get<\/span> <span class=\"hljs-keyword\">update<\/span>\r\nsudo apt-<span class=\"hljs-keyword\">get<\/span> <span class=\"hljs-keyword\">upgrade<\/span><\/code><\/pre>\n<h3><span id=\"62_nouveau\">2 \u7981\u7528nouveau\u9a71\u52a8<\/span><\/h3>\n<pre class=\"pure-highlightjs\"><code class=\"hljs nginx\"><span class=\"hljs-attribute\">sudo<\/span> vim \/etc\/modprobe.d\/blacklist.conf  <span class=\"hljs-comment\">#\u672b\u5c3e\u6dfb\u52a0\u5982\u4e0b\u51e0\u884c\uff1a <\/span>\r\n  blacklist vga16fb\r\n  blacklist nouveau\r\n  blacklist rivafb\r\n  blacklist rivatv\r\n  blacklist nvidiafb\r\n  blacklist lbm-nouveau\r\n    options nouveau modeset=<span class=\"hljs-number\">0<\/span>\r\n    alias nouveau <span class=\"hljs-literal\">off<\/span>\r\n    alias lbm-nouveau <span class=\"hljs-literal\">off<\/span>\r\n\r\nsudo update-initramfs -u\r\nreboot\r\n\r\nlsmod | grep nouveau\u00a0 \u00a0<span class=\"hljs-comment\">#\u67e5\u4e0d\u5230\u8868\u793anouveau\u5df2\u88ab\u5c4f\u853d<\/span><\/code><\/pre>\n<h3><span id=\"63_RTX2070\">3 \u5b89\u88c5\u663e\u5361RTX2070\u9a71\u52a8<\/span><\/h3>\n<pre class=\"pure-highlightjs\"><code class=\"hljs cs\"><span class=\"hljs-meta\"># \u5b89\u88c5\u4e4b\u524d\u5148\u5378\u8f7d\u5df2\u7ecf\u5b58\u5728\u7684\u9a71\u52a8\u7248\u672c<\/span>\r\nsudo apt-<span class=\"hljs-keyword\">get<\/span> remove --purge nvidia* libnvidia*<span class=\"hljs-meta\">\r\nsudo reboot\r\n# \u4f9d\u8d56\u9879<\/span>\r\nsudo apt-<span class=\"hljs-keyword\">get<\/span> install freeglut3-dev build-essential libx11-dev libxmu-dev libxi-dev libgl1-mesa-glx libglu1-mesa libglu1-mesa-dev\r\n<span class=\"hljs-meta\">#\u6dfb\u52a0Graphic Drivers PPA<\/span>\r\nsudo add-apt-repository ppa:graphics-drivers\/ppa\r\nsudo apt-<span class=\"hljs-keyword\">get<\/span> update\r\nubuntu-drivers devices  <span class=\"hljs-meta\">#\u5bfb\u627e\u5408\u9002\u7684\u9a71\u52a8\u7248\u672c\r\n#\u5b89\u88c5535\u7248\u672c\u9a71\u52a8<\/span>\r\nsudo apt-<span class=\"hljs-keyword\">get<\/span> install nvidia-driver<span class=\"hljs-number\">-535<\/span>\r\ncat \/proc\/driver\/nvidia\/version\r\n    NVRM version: NVIDIA UNIX x86_64 Kernel Module 535.216.01 Tue Sep 17 16:54:04 UTC 2024\r\n    GCC version: gcc version 9.4.0 (Ubuntu 9.4.0-1ubuntu1~20.04.2) \r\n\r\n<span class=\"hljs-meta\"># \u4e5f\u53ef\u4ee5\u8fdb\u5165nvidia\u5b98\u7f51 https:\/\/www.geforce.cn\/drivers\u4e0b\u8f7d\u5bf9\u5e94rtx2070\u663e\u5361\u7684\u9a71\u52a8\u7a0b\u5e8f\uff0c\u4e0b\u8f7d\u540e\u7684\u6587\u4ef6\u683c\u5f0f\u4e3arun<\/span>\r\n# sudo sh .\/NVIDIA-Linux-x86_64<span class=\"hljs-number\">-535.216<\/span>.run \u2013no-opengl-files\u00a0 \u00a0 <span class=\"hljs-meta\">#\u4e00\u8def\u56de\u8f66\u5373\u53ef<\/span>\r\n\r\nreboot  <span class=\"hljs-meta\">#\u91cd\u542f<\/span>\r\n\r\nnvidia-smi\u00a0 <span class=\"hljs-meta\">#\u68c0\u67e5\u663e\u5361\u9a71\u52a8\u662f\u5426\u6b63\u5e38<\/span><\/code><\/pre>\n<p><a href=\"https:\/\/yanjingang.com\/blog\/wp-content\/uploads\/2024\/01\/nvidia-ubuntu20.png\"><img loading=\"lazy\" class=\"alignnone size-full wp-image-8249\" src=\"https:\/\/yanjingang.com\/blog\/wp-content\/uploads\/2024\/01\/nvidia-ubuntu20.png\" alt=\"\" width=\"808\" height=\"403\" srcset=\"https:\/\/yanjingang.com\/blog\/wp-content\/uploads\/2024\/01\/nvidia-ubuntu20.png 808w, https:\/\/yanjingang.com\/blog\/wp-content\/uploads\/2024\/01\/nvidia-ubuntu20-300x150.png 300w, https:\/\/yanjingang.com\/blog\/wp-content\/uploads\/2024\/01\/nvidia-ubuntu20-768x383.png 768w, https:\/\/yanjingang.com\/blog\/wp-content\/uploads\/2024\/01\/nvidia-ubuntu20-624x311.png 624w\" sizes=\"(max-width: 808px) 100vw, 808px\" \/><\/a><\/p>\n<h1><span id=\"6ubuntugpu\">\u4e09\u3001\u5b89\u88c5CUDA\u3001cuDNN<\/span><\/h1>\n<h3>1. \u786e\u8ba4\u5339\u914d\u7684<span id=\"6ubuntugpu\">CUDA\u3001cuDNN\u7248\u672c<\/span><\/h3>\n<h6>1.1 \u786e\u8ba4\u64cd\u4f5c\u7cfb\u7edf\u3001GCC\u7248\u672c<\/h6>\n<pre class=\"pure-highlightjs\"><code class=\"\">$ cat \/etc\/issue\r\n    Ubuntu 20.04.6 LTS\r\n$ gcc --version\r\n    gcc (Ubuntu 9.4.0-1ubuntu1~20.04.2) 9.4.0<\/code><\/pre>\n<h6>1.2 \u786e\u8ba4PyTorch\/Tensorflow\u4e0eCUDA\u7248\u672c\u7684\u5173\u7cfb<\/h6>\n<p><a href=\"https:\/\/pytorch.org\/get-started\/previous-versions\/\" target=\"_blank\" rel=\"noopener noreferrer\">PyTorch\u4e0eCUDA\u7248\u672c\u5bf9\u5e94\u5173\u7cfb<\/a><\/p>\n<ul>\n<li>\u622a\u81f3\u5230\u4eca\u5929\u6700\u65b0\u7684<span style=\"color: #ff0000;\">pytorch<\/span>\u7248\u672c<span style=\"color: #ff0000;\">v2.1.1<\/span>\uff0c\u652f\u6301CUDA v12.1\u548c<span style=\"color: #ff0000;\">v11.8<\/span>\u7248\u672c\u3002<\/li>\n<\/ul>\n<p><a href=\"https:\/\/tensorflow.google.cn\/install\/source#linux\" target=\"_blank\" rel=\"noopener noreferrer\">Tensorflow\u7248\u672c\u4e0eCUDA\/CUDNN \u5bf9\u5e94\u5173\u7cfb<\/a><\/p>\n<ul>\n<li>\u622a\u81f3\u5230\u4eca\u5929\u6700\u65b0\u7684tensorflow-gpu<span style=\"color: #333333;\">\u7248\u672cv2.15.0\uff0c\u652f\u6301<\/span>CUDA v12.2\u00a0+ cuDNN v8.9\u3002<\/li>\n<li><span style=\"color: #ff0000;\">tensorflow<\/span>-gpu\u7248\u672c<span style=\"color: #ff0000;\">v2.14.0<\/span>\uff0c\u652f\u6301CUDA <span style=\"color: #ff0000;\">v11.8<\/span> + cuDNN <span style=\"color: #ff0000;\">v8.7<\/span>\u3002<\/li>\n<\/ul>\n<p>\u7efc\u5408\u6765\u770b\uff0c\u517c\u5bb9\u6700\u65b0PyTorch\/Tensorflow\u7248\u672c\u7684CUDA\u7248\u672c\u7ec4\u5408\u662f\uff1a<\/p>\n<ul>\n<li><span style=\"color: #ff0000;\">CUDA v11.8 + cuDNN v8.7<\/span><\/li>\n<\/ul>\n<h6>1.3 <strong>\u786e\u8ba4CUDA\u7248\u672c\u4e0eUbuntu\u3001GCC\u7248\u672c\u5173\u7cfb<\/strong><\/h6>\n<p><a href=\"https:\/\/docs.nvidia.com\/cuda\/archive\/11.8.0\/cuda-installation-guide-linux\/index.html\" target=\"_blank\" rel=\"noopener\">CUDA11.8\u7248\u672c\u4e0eUbuntu\/GCC\u7248\u672c\u5bf9\u5e94\u5173\u7cfb<\/a><\/p>\n<p>Table 1. Native Linux Distribution Support in CUDA\u00a0<span class=\"keyword\">11.8<\/span><\/p>\n<table class=\"table\" border=\"1\" summary=\"\" frame=\"border\" rules=\"all\" cellspacing=\"0\" cellpadding=\"4\">\n<tbody class=\"tbody\">\n<tr class=\"row\">\n<td class=\"entry\" colspan=\"1\" rowspan=\"1\" headers=\"d117e166\" valign=\"top\" width=\"NaN%\"><strong>Distribution<\/strong><\/td>\n<td class=\"entry\" colspan=\"1\" rowspan=\"1\" headers=\"d117e169\" align=\"center\" valign=\"middle\" width=\"NaN%\"><strong>Kernel<sup class=\"ph sup\">1<\/sup><\/strong><\/td>\n<td class=\"entry\" colspan=\"1\" rowspan=\"1\" headers=\"d117e174\" align=\"center\" valign=\"middle\" width=\"NaN%\"><strong>Default GCC<\/strong><\/td>\n<td class=\"entry\" colspan=\"1\" rowspan=\"1\" headers=\"d117e177\" align=\"center\" valign=\"middle\" width=\"NaN%\"><strong>GLIB<\/strong>C<\/td>\n<\/tr>\n<tr class=\"row\">\n<td class=\"entry\" colspan=\"1\" rowspan=\"1\" headers=\"d117e166\" valign=\"top\" width=\"NaN%\">x86_64<\/td>\n<td class=\"entry\" colspan=\"1\" rowspan=\"1\" headers=\"d117e169\" align=\"center\" valign=\"middle\" width=\"NaN%\"><\/td>\n<td class=\"entry\" colspan=\"1\" rowspan=\"1\" headers=\"d117e174\" align=\"center\" valign=\"middle\" width=\"NaN%\"><\/td>\n<td class=\"entry\" colspan=\"1\" rowspan=\"1\" headers=\"d117e177\" align=\"center\" valign=\"middle\" width=\"NaN%\"><\/td>\n<\/tr>\n<tr class=\"row\">\n<td class=\"entry\" colspan=\"1\" rowspan=\"1\" headers=\"d117e166\" valign=\"top\" width=\"NaN%\">Ubuntu 22.04 LTS<\/td>\n<td class=\"entry\" colspan=\"1\" rowspan=\"1\" headers=\"d117e169\" align=\"center\" valign=\"middle\" width=\"NaN%\">5.15.0-25<\/td>\n<td class=\"entry\" colspan=\"1\" rowspan=\"1\" headers=\"d117e174\" align=\"center\" valign=\"middle\" width=\"NaN%\">11.2.0<\/td>\n<td class=\"entry\" colspan=\"1\" rowspan=\"1\" headers=\"d117e177\" align=\"center\" valign=\"middle\" width=\"NaN%\">2.35<\/td>\n<\/tr>\n<tr class=\"row\">\n<td class=\"entry\" colspan=\"1\" rowspan=\"1\" headers=\"d117e166\" valign=\"top\" width=\"NaN%\"><strong>Ubuntu 20.04.z (z &lt;= 4) LTS<\/strong><\/td>\n<td class=\"entry\" colspan=\"1\" rowspan=\"1\" headers=\"d117e169\" align=\"center\" valign=\"middle\" width=\"NaN%\"><strong>5.13.0-30<\/strong><\/td>\n<td class=\"entry\" colspan=\"1\" rowspan=\"1\" headers=\"d117e174\" align=\"center\" valign=\"middle\" width=\"NaN%\"><strong>9.3.0<\/strong><\/td>\n<td class=\"entry\" colspan=\"1\" rowspan=\"1\" headers=\"d117e177\" align=\"center\" valign=\"middle\" width=\"NaN%\"><strong>2.31<\/strong><\/td>\n<\/tr>\n<tr class=\"row\">\n<td class=\"entry\" colspan=\"1\" rowspan=\"1\" headers=\"d117e166\" valign=\"top\" width=\"NaN%\">Ubuntu 18.04.z (z &lt;= 6) LTS<\/td>\n<td class=\"entry\" colspan=\"1\" rowspan=\"1\" headers=\"d117e169\" align=\"center\" valign=\"middle\" width=\"NaN%\">5.4.0-89<\/td>\n<td class=\"entry\" colspan=\"1\" rowspan=\"1\" headers=\"d117e174\" align=\"center\" valign=\"middle\" width=\"NaN%\">7.5.0<\/td>\n<td class=\"entry\" colspan=\"1\" rowspan=\"1\" headers=\"d117e177\" align=\"center\" valign=\"middle\" width=\"NaN%\">2.27<\/td>\n<\/tr>\n<\/tbody>\n<\/table>\n<p>\u53ef\u4ee5\u770b\u5230CUDA11.8\u5728Ubuntu20.04\u4e0b\u9700\u8981GCC &gt;=9.3.0\uff0c\u7b26\u5408\u5f53\u524d\u73af\u5883\u8981\u6c42\u3002<\/p>\n<p><a href=\"https:\/\/docs.nvidia.com\/cuda\/cuda-toolkit-release-notes\/index.html\" target=\"_blank\" rel=\"noopener noreferrer\">Nvidia\u663e\u5361\u9a71\u52a8\u4e0eCUDA\u7248\u672c\u5bf9\u5e94\u5173\u7cfb<\/a><\/p>\n<table id=\"id3\" class=\"small table-no-stripes docutils align-default\">\n<caption><span class=\"caption-number\">Table 2\u00a0<\/span><span class=\"caption-text\">CUDA Toolkit and Minimum Required Driver Version for CUDA Minor Version Compatibility<\/span><a class=\"headerlink\" title=\"Permalink to this table\" href=\"https:\/\/docs.nvidia.com\/cuda\/cuda-toolkit-release-notes\/index.html#id3\">\uf0c1<\/a><\/caption>\n<colgroup>\n<col \/>\n<col \/>\n<col \/><\/colgroup>\n<thead>\n<tr class=\"row-odd\">\n<th class=\"head\">CUDA Toolkit<\/th>\n<th class=\"head\" colspan=\"2\">Minimum Required Driver Version for CUDA Minor Version Compatibility*<\/th>\n<\/tr>\n<\/thead>\n<tbody>\n<tr class=\"row-even\">\n<td><\/td>\n<td>Linux x86_64 Driver Version<\/td>\n<td>Windows x86_64 Driver Version<\/td>\n<\/tr>\n<tr class=\"row-odd\">\n<td>CUDA 11.8.x<\/td>\n<td>&gt;=450.80.02<\/td>\n<td>&gt;=452.39<\/td>\n<\/tr>\n<\/tbody>\n<\/table>\n<p>cat \/proc\/driver\/nvidia\/version\u00a0 \u53ef\u4ee5\u770b\u5230\u6211\u4eec\u5b89\u88c5\u7684\u9a71\u52a8\u662f535.154.05\uff0c\u6ee1\u8db3CUDA11.8.x\u7684\u9a71\u52a8\u4f9d\u8d56\u6761\u4ef6\u3002<\/p>\n<h6>1.4 \u786e\u8ba4TensorRT\u4e0eCUDA\u7248\u672c\u7684\u5173\u7cfb<\/h6>\n<p>\u67e5\u770b\u6700\u65b0<a href=\"https:\/\/docs.nvidia.com\/deeplearning\/tensorrt\/archives\/\" target=\"_blank\" rel=\"noopener\">TensorRT<\/a>\u7248\u672c\u7684<a href=\"https:\/\/docs.nvidia.com\/deeplearning\/tensorrt\/archives\/tensorrt-853\/install-guide\/index.html#installing-tar\" target=\"_blank\" rel=\"noopener\">NVIDIA TensorRT Installation Guide<\/a>\u3002\u7ecf\u8fc7\u5bf9\u6bd4\uff0c\u540c\u65f6\u652f\u6301 CUDA v11.8 + cuDNN v8.7 \u7684TensorRT\u6700\u65b0\u7248\u672c\u4e3a<span style=\"color: #ff0000;\">v8.5.3<\/span>\u3002<\/p>\n<p><a href=\"https:\/\/yanjingang.com\/blog\/wp-content\/uploads\/2024\/02\/tensor-rt.png\"><img loading=\"lazy\" class=\"alignnone size-full wp-image-8284\" src=\"https:\/\/yanjingang.com\/blog\/wp-content\/uploads\/2024\/02\/tensor-rt.png\" alt=\"\" width=\"1016\" height=\"154\" srcset=\"https:\/\/yanjingang.com\/blog\/wp-content\/uploads\/2024\/02\/tensor-rt.png 1016w, https:\/\/yanjingang.com\/blog\/wp-content\/uploads\/2024\/02\/tensor-rt-300x45.png 300w, https:\/\/yanjingang.com\/blog\/wp-content\/uploads\/2024\/02\/tensor-rt-768x116.png 768w, https:\/\/yanjingang.com\/blog\/wp-content\/uploads\/2024\/02\/tensor-rt-624x95.png 624w\" sizes=\"(max-width: 1016px) 100vw, 1016px\" \/><\/a><\/p>\n<h3><span id=\"64_CUDA100\">2. \u5b89\u88c5CUDA 11.8<\/span><\/h3>\n<p>\u4ece<a href=\"https:\/\/developer.nvidia.com\/cuda-11-8-0-download-archive\" target=\"_blank\" rel=\"noopener noreferrer\">https:\/\/developer.nvidia.com\/cuda-11-8-0-download-archive<\/a> \u4e0b\u8f7dcuda-11.18\uff0c\u9009\u62e9Linux + x86_64 + Ubuntu + 20.04 + runfile(local)\uff0c<\/p>\n<p><a href=\"https:\/\/yanjingang.com\/blog\/wp-content\/uploads\/2024\/01\/cuda11.8.png\"><img loading=\"lazy\" class=\"alignnone size-large wp-image-8251\" src=\"https:\/\/yanjingang.com\/blog\/wp-content\/uploads\/2024\/01\/cuda11.8-1024x608.png\" alt=\"\" width=\"625\" height=\"371\" srcset=\"https:\/\/yanjingang.com\/blog\/wp-content\/uploads\/2024\/01\/cuda11.8-1024x608.png 1024w, https:\/\/yanjingang.com\/blog\/wp-content\/uploads\/2024\/01\/cuda11.8-300x178.png 300w, https:\/\/yanjingang.com\/blog\/wp-content\/uploads\/2024\/01\/cuda11.8-768x456.png 768w, https:\/\/yanjingang.com\/blog\/wp-content\/uploads\/2024\/01\/cuda11.8-624x370.png 624w, https:\/\/yanjingang.com\/blog\/wp-content\/uploads\/2024\/01\/cuda11.8.png 1308w\" sizes=\"(max-width: 625px) 100vw, 625px\" \/><\/a><\/p>\n<pre class=\"pure-highlightjs\"><code class=\"hljs sql\">#\u4e0b\u8f7d\u5b89\u88c5cuda11.8    \uff08\u624b\u5de5\u5b89\u88c5\uff0c\u522b\u7528sudo apt <span class=\"hljs-keyword\">install<\/span> nvidia-cuda-toolkit\u88c5\uff0c\u7248\u672c\u4f1a\u5f88\u65e7\uff09\r\nwget -c https:\/\/developer.download.nvidia.com\/compute\/cuda\/11.8.0\/local_installers\/cuda_11.8.0_520.61.05_linux.run\r\nchmod +x cuda_11.8.0_520.61.05_linux.run\r\nsudo sh cuda_11.8.0_520.61.05_linux.run\r\n\r\n  <span class=\"hljs-keyword\">Do<\/span> you <span class=\"hljs-keyword\">accept<\/span> the previously <span class=\"hljs-keyword\">read<\/span> EULA?\r\n  <span class=\"hljs-keyword\">accept<\/span>\/decline\/quit: <span class=\"hljs-keyword\">accept<\/span>\r\n\r\n  <span class=\"hljs-keyword\">Install<\/span> NVIDIA Accelerated Graphics Driver <span class=\"hljs-keyword\">for<\/span> Linux-x86_64 520.61.05?\r\n  (y)es\/(<span class=\"hljs-keyword\">n<\/span>)o\/(q)uit: <span class=\"hljs-keyword\">n<\/span>    #\u524d\u8fb9\u5df2\u7ecf\u88c5\u8fc7\u663e\u5361\u9a71\u52a8\u4e86\uff0c\u8fd9\u91cc\u4e0d\u8981\u91cd\u590d\u5b89\u88c5\r\n\r\n  <span class=\"hljs-keyword\">Install<\/span> the CUDA <span class=\"hljs-number\">11.8<\/span> Toolkit?\r\n  (y)es\/(<span class=\"hljs-keyword\">n<\/span>)o\/(q)uit: y\r\n\r\n  Enter Toolkit Location\r\n   [ <span class=\"hljs-keyword\">default<\/span> <span class=\"hljs-keyword\">is<\/span> \/usr\/<span class=\"hljs-keyword\">local<\/span>\/cuda<span class=\"hljs-number\">-11.8<\/span> ]: \r\n\r\n  Installing the CUDA Toolkit <span class=\"hljs-keyword\">in<\/span> \/usr\/<span class=\"hljs-keyword\">local<\/span>\/cuda<span class=\"hljs-number\">-11.8<\/span> ...\r\n  Installing the CUDA Samples <span class=\"hljs-keyword\">in<\/span> \/home\/<span class=\"hljs-keyword\">work<\/span> ...\r\n  Copying samples <span class=\"hljs-keyword\">to<\/span> \/home\/<span class=\"hljs-keyword\">work<\/span>\/NVIDIA_CUDA<span class=\"hljs-number\">-11.8<\/span>_Samples <span class=\"hljs-keyword\">now<\/span>...\r\n  Finished copying samples.\r\n<\/code><code class=\"hljs http\">\r\n\r\n<span class=\"cpp\"><span class=\"hljs-meta\">#\u67e5\u770bcuda\u7248\u672c<\/span><\/span><\/code><code class=\"prism language-powershell has-numbering\">\r\ncd \/usr\/local\/cuda-11.8\/bin\r\n.\/nvcc --version\r\n  nvcc: NVIDIA (R) Cuda compiler driver\r\n  Copyright (c) 2005-2022 NVIDIA Corporation\r\n  Built on Wed_Sep_21_10:33:58_PDT_2022\r\n  Cuda compilation tools, release 11.8, V11.8.89\r\n  Build cuda_11.8.r11.8\/compiler.31833905_0\r\n\r\n#\u6dfb\u52a0\u73af\u5883\u53d8\u91cf\r\nvim ~\/.bashrc\r\n  export CUDA_HOME=\/usr\/local\/cuda-11.8\r\n  export CUDA_PATH=${CUDA_HOME}\r\n  export CUDA_TOOLKIT_ROOT_DIR=${CUDA_HOME}\r\n  export PATH=${CUDA_HOME}\/bin:${PATH}\r\n  export LD_LIBRARY_PATH=${CUDA_HOME}\/lib64:${LD_LIBRARY_PATH}\r\nsource ~\/.bashrc<\/code><\/pre>\n<p>\u6b64\u65f6\u987a\u4fbf\u518d\u662f\u4e0bnvidia-smi\uff0c\u770b\u770bgpu\u9a71\u52a8\u662f\u5426\u8fd8\u80fd\u6b63\u5e38\u5de5\u4f5c\u3002<\/p>\n<h3><span id=\"65_cuDNN74\">3. \u5b89\u88c5cuDNN 8.7<\/span><\/h3>\n<p><a href=\"https:\/\/developer.nvidia.com\/rdp\/cudnn-archive\" target=\"_blank\" rel=\"noopener noreferrer\">https:\/\/developer.nvidia.com\/rdp\/cudnn-archive<\/a><\/p>\n<p>\u6ce8\u518c\u8d26\u53f7\uff0c\u4e0b\u8f7d\u5bf9\u5e94cudnn\u7248\u672c\uff0c\u6211\u9009\u62e9\u7684 <a class=\"\" href=\"https:\/\/developer.nvidia.com\/rdp\/cudnn-archive#a-collapse870-118\" data-parent=\"#accordion\" data-toggle=\"collapse\" aria-expanded=\"true\">Download cuDNN v8.7.0 (November 28th, 2022), for CUDA 11.x <\/a>\u4e0b\u7684\u00a0<a href=\"https:\/\/developer.nvidia.com\/downloads\/c118-cudnn-linux-8664-87084cuda11-archivetarz\" target=\"_blank\" rel=\"noopener\">Local Installer for Linux x86_64 (Tar)<\/a><\/p>\n<pre class=\"pure-highlightjs\"><code class=\"hljs perl\"><span class=\"hljs-comment\">#\u4e0b\u8f7dCUDNN<\/span>\r\nwget -c -O cudnn-linux-x86_64-8.7.0.84_cuda11-archive.tar.xz \"https:\/\/developer.download.nvidia.com\/compute\/cudnn\/secure\/8.7.0\/local_installers\/11.8\/cudnn-linux-x86_64-8.7.0.84_cuda11-archive.tar.xz?_yhgesiq1TihsP6a_eh8LBpOblhF9j0gWC9-tOYrqmrXtd3VzJ9AhgaKQc3Hrel73ti3r-i4xD1O9bqFfXDB_xBws9P9RrDwr2W5AJQ8_oUCRoy_0h_b0BpdWfCHlOrS0Wez8umHVfC96DN0ltsuxnaD3ZfIcRnqNCjCyh5SzSf06ZdFtqzyYtHLW6Wgxd3jCx1qPhc8cy1D3PChIpdDHnc=&amp;t=eyJscyI6IndlYnNpdGUiLCJsc2QiOiJkZXZlbG9wZXIubnZpZGlhLmNvbS9jdWRhLTExLjgtZG93bmxvYWQtYXJjaGl2ZSJ9\"\r\n<span class=\"hljs-comment\">#\u89e3\u538b\u540e\u7684\u6587\u4ef6\u5939\u540d\u79f0\u4e3acuda ,\u5c06\u5bf9\u5e94\u6587\u4ef6\u590d\u5236\u5230 \/usr\/local\u4e2d\u7684cuda-<span class=\"hljs-number\">11.8<\/span>\u76ee\u5f55\u5185<\/span>\r\ntar xvf cudnn-linux-x86_64-8.7.0.84_cuda11-archive.tar.xz\r\n<span class=\"hljs-comment\"># \u590d\u5236cudnn\u5934\u6587\u4ef6\u548clib<span class=\"hljs-regexp\">\u5e93<\/span><\/span>\r\ncd cudnn-linux-x86_64-8.7.0.84_cuda11-archive\r\nsudo cp .\/include\/* <span class=\"hljs-regexp\">\/usr\/local<\/span><span class=\"hljs-regexp\">\/cuda-<span class=\"hljs-number\">11.8<\/span>\/include<\/span><span class=\"hljs-regexp\">\/\r\nsudo cp .\/<\/span><span class=\"hljs-regexp\">lib\/<\/span>* <span class=\"hljs-regexp\">\/usr\/local<\/span><span class=\"hljs-regexp\">\/cuda-<span class=\"hljs-number\">11.8<\/span>\/lib<\/span>64\/\r\n<\/code><\/pre>\n<h3>4.<span style=\"font-size: 1.5rem;\">\u5b89\u88c5<\/span>TensorRT 8.5.3<\/h3>\n<p>\u5b98\u65b9\u8bf4\u660e\uff1a<a href=\"https:\/\/docs.nvidia.com\/deeplearning\/tensorrt\/archives\/tensorrt-853\/install-guide\/index.html#installing-tar\" target=\"_blank\" rel=\"noopener\">Installation Guide<\/a><\/p>\n<p>\u4e0b\u8f7d\u5730\u5740\uff1a<a href=\"https:\/\/developer.nvidia.com\/nvidia-tensorrt-8x-download\" target=\"_blank\" rel=\"noopener\">NVIDIA TensorRT 8.x Download<\/a><\/p>\n<ul>\n<li style=\"list-style-type: none;\">\n<ul>\n<li><a href=\"https:\/\/developer.nvidia.com\/downloads\/compute\/machine-learning\/tensorrt\/secure\/8.5.3\/tars\/TensorRT-8.5.3.1.Linux.x86_64-gnu.cuda-11.8.cudnn8.6.tar.gz\">TensorRT 8.5 GA Update 2 for Linux x86_64 and CUDA 11.0, 11.1, 11.2, 11.3, 11.4, 11.5, 11.6, 11.7 and 11.8 TAR Package<\/a><\/li>\n<\/ul>\n<\/li>\n<\/ul>\n<p>\u5b89\u88c5\uff1a<\/p>\n<pre class=\"pure-highlightjs\"><code class=\"\"># Unpack\r\nversion=\"8.5.3.1\"\r\ntar zxvf TensorRT-${version}.Linux.x86_64-gnu.cuda-11.8.cudnn8.6.tar.gz\r\n\r\n# add LD_LIBRARY_PATH\r\ninstallpath=\/home\/work\/tools\/gpu\/TensorRT-${version}\r\nexport LD_LIBRARY_PATH=$LD_LIBRARY_PATH:${installpath}\/lib\r\n\r\n# Python wheel\r\npython3 -m pip install ${installpath}\/python\/tensorrt-*-cp38-none-linux_x86_64.whl\r\npython3 -m pip install ${installpath}\/graphsurgeon\/graphsurgeon-0.4.6-py2.py3-none-any.whl\r\npython3 -m pip install ${installpath}\/uff\/uff-0.6.9-py2.py3-none-any.whl\r\npython3 -m pip install ${installpath}\/onnx_graphsurgeon\/onnx_graphsurgeon-0.3.12-py2.py3-none-any.whl\r\n\r\n# test\r\ncd ${installpath}\/samples\/python\/yolov3_onnx\r\npip3 install -r requirements.txt\r\nwget https:\/\/github.com\/pjreddie\/darknet\/raw\/f86901f6177dfc6116360a13cc06ab680e0c86b0\/data\/dog.jpg\r\nwget https:\/\/raw.githubusercontent.com\/pjreddie\/darknet\/f86901f6177dfc6116360a13cc06ab680e0c86b0\/cfg\/yolov3.cfg\r\nwget -c -O yolov3.weights https:\/\/pjreddie.com\/media\/files\/yolov3.weights\r\npython3 yolov3_to_onnx.py<\/code><\/pre>\n<h3>5. \u68c0\u67e5\u7248\u672c<\/h3>\n<pre class=\"pure-highlightjs\"><code class=\"\">$ nvcc -V\r\n$ dpkg -l | grep -i cudnn\r\n$ dpkg-query -W tensorrt<\/code><\/pre>\n<h3><span style=\"font-size: 1.5rem;\">6.\u5b89\u88c5Torch\/Tensorflow\uff0c\u9a8c\u8bc1CUDA\u662f\u5426\u751f\u6548<\/span><\/h3>\n<pre class=\"pure-highlightjs\"><code class=\"hljs python\"># pip\u5b89\u88c5pytorch\/tensorflow\r\nsudo apt install python3-pip\r\npip3 install --upgrade pip\r\npip3 install torch tensorflow tensorrt\r\n\r\n# \u7f16\u5199pytorch\/tensorflow\u6d4b\u8bd5\u811a\u672c\r\nvim test_gpu.py\r\n  # test torch\r\n  <span class=\"hljs-keyword\">import<\/span> torch\r\n  print(\"Torch CUDA: \", torch.cuda.is_available())\r\n  device = torch.device(<span class=\"hljs-string\">\"cuda:0\"<\/span>)\r\n  print(device)\r\n  print(torch.cuda.get_device_name(<span class=\"hljs-number\">0<\/span>))\r\n  print(torch.rand(<span class=\"hljs-number\">3<\/span>,<span class=\"hljs-number\">3<\/span>).cuda()) \r\n  print(torch.rand(<span class=\"hljs-number\">5<\/span>,<span class=\"hljs-number\">5<\/span>).cuda() + torch.rand(<span class=\"hljs-number\">5<\/span>,<span class=\"hljs-number\">5<\/span>).cuda())\r\n  # test tensorflow\r\n  <span class=\"hljs-keyword\">import<\/span> tensorflow <span class=\"hljs-keyword\">as<\/span> tf\r\n  print(\"Tensorflow GPU: \", tf.test.is_gpu_available())\r\n\r\n# \u6d4b\u8bd5GPU\u662f\u5426\u6210\u6548\r\npython test_gpu.py\r\n  Torch CUDA: <span class=\"hljs-keyword\">True<\/span>\r\n  cuda:<span class=\"hljs-number\">0<\/span>\r\n  GeForce RTX <span class=\"hljs-number\">2070<\/span>\r\n  tensor([[<span class=\"hljs-number\">0.9530<\/span>, <span class=\"hljs-number\">0.4746<\/span>, <span class=\"hljs-number\">0.9819<\/span>],\r\n          [<span class=\"hljs-number\">0.7192<\/span>, <span class=\"hljs-number\">0.9427<\/span>, <span class=\"hljs-number\">0.6768<\/span>],\r\n          [<span class=\"hljs-number\">0.8594<\/span>, <span class=\"hljs-number\">0.9490<\/span>, <span class=\"hljs-number\">0.6551<\/span>]], device=<span class=\"hljs-string\">'cuda:0'<\/span>)\r\n\r\n  Created device \/device:GPU:0 with 6412 MB memory: -&gt; device: 0, name: NVIDIA GeForce RTX 2070, pci bus id: 0000:01:00.0, compute capability: 7.5\r\n  Tensorflow GPU: <span class=\"hljs-keyword\">True<\/span>\r\n\r\nnvidia-smi -l   <span class=\"hljs-comment\">#\u67e5\u770bgpu\u5360\u7528\u8fdb\u7a0b\u60c5\u51b5<\/span><\/code><\/pre>\n<p>&nbsp;<\/p>\n<p>&nbsp;<\/p>\n<p>yan 24.2.1 0:18<\/p>\n<p>&nbsp;<\/p>\n<p>\u53c2\u8003\uff1a<\/p>\n<p><a href=\"https:\/\/blog.csdn.net\/weixin_40779727\/article\/details\/90453345\" target=\"_blank\" rel=\"noopener noreferrer\">\u53cc\u7cfb\u7edf\u5b89\u88c5Ubantu18.04 + RTX2070 + CUDA10.1 + Cudnn7.5.1+anaconda3+tensorflow-gpu<\/a><\/p>\n<p><a href=\"https:\/\/blog.csdn.net\/s717597589\/article\/details\/79117112\" target=\"_blank\" rel=\"noopener noreferrer\">win10\u4e0b\u5b89\u88c5Ubuntu16.04\u53cc\u7cfb\u7edf<\/a><\/p>\n<p><a href=\"https:\/\/www.cnblogs.com\/MartinLwx\/p\/10409230.html\" target=\"_blank\" rel=\"noopener noreferrer\">Linux\uff08Ubuntu\uff09\u6362apt-get\u6e90<\/a><\/p>\n<p><a href=\"https:\/\/blog.csdn.net\/zwliang98\/article\/details\/83546788\" target=\"_blank\" rel=\"noopener noreferrer\">pip\/pip3\u66f4\u6362\u56fd\u5185\u955c\u50cf\u6e90<\/a><\/p>\n<p><a href=\"https:\/\/blog.csdn.net\/qq_35494379\/article\/details\/105603832\" target=\"_blank\" rel=\"noopener noreferrer\">Ubuntu18.04\u4e0b\u5b89\u88c5Pytorch-GPU<\/a><\/p>\n<p><a href=\"https:\/\/docs.nvidia.com\/cuda\/cuda-toolkit-release-notes\/index.html\" target=\"_blank\" rel=\"noopener noreferrer\">Nvidia\u663e\u5361\u9a71\u52a8\u4e0eCUDA\u7248\u672c\u5bf9\u5e94\u5173\u7cfb<\/a><\/p>\n<p><a href=\"https:\/\/docs.nvidia.com\/cuda\/cuda-installation-guide-linux\/index.html\" target=\"_blank\" rel=\"noopener noreferrer\">CUDA\u7248\u672c\u4e0eUbuntu \u7248\u672c\uff0c\u4ee5\u53caGCC\u7248\u672c\u5bf9\u5e94\u5173\u7cfb<\/a><\/p>\n<p><a href=\"https:\/\/tensorflow.google.cn\/install\/source#linux\" target=\"_blank\" rel=\"noopener noreferrer\">Tensorflow\u7248\u672c\u4e0eCUDA\/CUDNN \u5bf9\u5e94\u5173\u7cfb<\/a><\/p>\n","protected":false},"excerpt":{"rendered":"<p>\u672c\u6587\u4e3b\u8981\u8bb0\u5f55\u534e\u7855\u7684GPU\u672c\uff08RTX2070\uff09\u5982\u4f55Ubuntu20.04 + Nvidia\u663e\u5361\u9a71\u52a8 + CUDA [&hellip;]<\/p>\n","protected":false},"author":1,"featured_media":0,"comment_status":"open","ping_status":"closed","sticky":false,"template":"","format":"standard","meta":[],"categories":[31,539],"tags":[736,737,1002,1004,1003,1340,1339,1001,855,1000],"_links":{"self":[{"href":"https:\/\/yanjingang.com\/blog\/index.php?rest_route=\/wp\/v2\/posts\/8246"}],"collection":[{"href":"https:\/\/yanjingang.com\/blog\/index.php?rest_route=\/wp\/v2\/posts"}],"about":[{"href":"https:\/\/yanjingang.com\/blog\/index.php?rest_route=\/wp\/v2\/types\/post"}],"author":[{"embeddable":true,"href":"https:\/\/yanjingang.com\/blog\/index.php?rest_route=\/wp\/v2\/users\/1"}],"replies":[{"embeddable":true,"href":"https:\/\/yanjingang.com\/blog\/index.php?rest_route=%2Fwp%2Fv2%2Fcomments&post=8246"}],"version-history":[{"count":3,"href":"https:\/\/yanjingang.com\/blog\/index.php?rest_route=\/wp\/v2\/posts\/8246\/revisions"}],"predecessor-version":[{"id":11189,"href":"https:\/\/yanjingang.com\/blog\/index.php?rest_route=\/wp\/v2\/posts\/8246\/revisions\/11189"}],"wp:attachment":[{"href":"https:\/\/yanjingang.com\/blog\/index.php?rest_route=%2Fwp%2Fv2%2Fmedia&parent=8246"}],"wp:term":[{"taxonomy":"category","embeddable":true,"href":"https:\/\/yanjingang.com\/blog\/index.php?rest_route=%2Fwp%2Fv2%2Fcategories&post=8246"},{"taxonomy":"post_tag","embeddable":true,"href":"https:\/\/yanjingang.com\/blog\/index.php?rest_route=%2Fwp%2Fv2%2Ftags&post=8246"}],"curies":[{"name":"wp","href":"https:\/\/api.w.org\/{rel}","templated":true}]}}