{"id":3864,"date":"2020-06-13T15:13:09","date_gmt":"2020-06-13T07:13:09","guid":{"rendered":"https:\/\/yanjingang.com\/blog\/?p=3864"},"modified":"2024-09-23T21:54:39","modified_gmt":"2024-09-23T13:54:39","slug":"%e5%8d%8e%e7%a1%95%e9%ad%94%e9%9c%b83%e5%ae%89%e8%a3%85win10-ubuntu18-4%e5%8f%8c%e7%b3%bb%e7%bb%9f","status":"publish","type":"post","link":"https:\/\/yanjingang.com\/blog\/?p=3864","title":{"rendered":"\u534e\u7855ROG\u5b89\u88c5ubuntu18.4+CUDA10.0+CUDNN7.4"},"content":{"rendered":"

\u7533\u8bf7\u4e86\u597d\u4e45\u7684\u9884\u7b97\u8001\u5a46\u7ec8\u4e8e\u6279\u4e86\uff0c\u5165\u624b\u534e\u7855\u7684GPU\u672c\uff08RTX2070\uff09\uff0c\u9ed8\u8ba4\u9884\u88c5\u4e86\u6b63\u7248win10\uff0c\u5e73\u65f6\u8fd8\u662f\u4e60\u60ef\u7528linux\uff0c\u672c\u6587\u8bb0\u5f55\u4e86\u5982\u4f55\u5b89\u88c5ubuntu18.4\u7684\u8fc7\u7a0b\u3002<\/p>\n

\"\"<\/a><\/p>\n

1.\u4e0b\u8f7dubuntu18.4\u7684iso\u955c\u50cf\u6587\u4ef6<\/h1>\n

\u81ea\u5df1\u53bb\u5b98\u7f51\u4e0b\u8f7d\uff1ahttps:\/\/www.ubuntu.com\/download\/desktop<\/p>\n

2.\u628aiso\u955c\u50cf\u70e7\u5f55\u5230u\u76d8\u91cc<\/h1>\n

\u63a8\u8350\u4f7f\u7528balenaEtcher<\/a>\uff0c\u5b83\u53ef\u4ee5\u5728 Linux\u3001Windows \u548c Mac OS \u4e0a\u8fd0\u884c\u3002<\/p>\n

3.\u51c6\u5907ubuntu\u78c1\u76d8<\/h1>\n

\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

4.\u91cd\u542f\u6309F2\u4fee\u6539bois\u8bbe\u7f6e<\/h1>\n

4.1 Secure Boot \u8bbe\u7f6e\u6210Disable<\/h3>\n

\u5426\u5219\u4f1a\u9ed1\u5c4f<\/p>\n

4.2 Boot\u542f\u52a8\u987a\u5e8f\u6539\u4e3aU\u76d8\u5728win10\u524d\u8fb9<\/h3>\n

4.3 STAT Model\u6539\u4e3aAHCI<\/h3>\n

\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

5.\u5b89\u88c5ubuntu<\/h1>\n

\u5982\u679c\u9047\u5230No EFI System Partition was found.this system will likely….. Go back ….\u65f6\uff0c\u9700\u8981\u65b0\u5efa\u4e00\u4e2a\u7528\u4e8eEFI\u7684\u5206\u533a\uff0c\u7ed9100M\u5c31\u53ef\u4ee5\u3002<\/p>\n

6.\u8fdb\u5165ubuntu\uff0c\u5b89\u88c5gpu\u9a71\u52a8<\/h1>\n

6.1 \u51c6\u5907\u5de5\u4f5c<\/h3>\n
sudo apt-get install vim-gtk openssh-server\r\n\r\nsudo vim \/etc\/default\/grub\r\n  GRUB_GFXMODE=1920x1080\r\n\r\nsudo update-grub\r\nsudo apt-get update\r\nsudo apt-get upgrade<\/code><\/pre>\n
6.2 \u7981\u7528nouveau\u9a71\u52a8<\/h3>\n
sudo vim \/etc\/modprobe.d\/blacklist.conf  #\u672b\u5c3e\u6dfb\u52a0\u5982\u4e0b\u51e0\u884c\uff1a \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=0\r\n    alias nouveau off\r\n    alias lbm-nouveau off\r\n\r\nsudo update-initramfs -u\r\nreboot\r\n\r\nlsmod | grep nouveau\u00a0 \u00a0#\u67e5\u4e0d\u5230\u8868\u793anouveau\u5df2\u88ab\u5c4f\u853d<\/code><\/pre>\n
6.3 \u5b89\u88c5\u663e\u5361RTX2070\u9a71\u52a8<\/h3>\n
sudo apt-get remove --purge nvidia*\u00a0 #\u5b89\u88c5\u4e4b\u524d\u5148\u5378\u8f7d\u5df2\u7ecf\u5b58\u5728\u7684\u9a71\u52a8\u7248\u672c\r\nsudo apt-get install freeglut3-dev build-essential libx11-dev libxmu-dev libxi-dev libgl1-mesa-glx libglu1-mesa libglu1-mesa-dev\u00a0 #\u4f9d\u8d56\u9879\r\nsudo add-apt-repository ppa:graphics-drivers\/ppa\u00a0 #\u6dfb\u52a0Graphic Drivers PPA\r\nsudo apt-get update\r\nubuntu-drivers devices  #\u5bfb\u627e\u5408\u9002\u7684\u9a71\u52a8\u7248\u672c\r\nsudo apt-get install nvidia-driver-440  #\u5b89\u88c5440\u7248\u672c\u9a71\u52a8\r\ncat \/proc\/driver\/nvidia\/version\r\n    NVRM version: NVIDIA UNIX x86_64 Kernel Module  440.82<\/span>\r\n\r\n#440\u7248\u672c\u9a71\u52a8\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\r\nsudo sh .\/NVIDIA-Linux-x86_64-440.82.run \u2013no-opengl-files\u00a0 \u00a0 #\u4e00\u8def\u56de\u8f66\u5373\u53ef\r\n\r\nreboot  #\u91cd\u542f\r\n\r\nnvidia-smi\u00a0 #\u68c0\u67e5\u663e\u5361\u9a71\u52a8\u662f\u5426\u6b63\u5e38<\/code><\/pre>\n
\"\"<\/a><\/p>\n
\u5230\u7cfb\u7edf\u8bbe\u7f6e\u7684\u201c\u8be6\u7ec6\u4fe1\u606f\u201d\u90e8\u5206\uff0c\u67e5\u770b\u4e00\u4e0b\u81ea\u5df1\u7684\u56fe\u5f62\u5361\u662f\u5426\u5df2\u7ecf\u662f\u521a\u624d\u7684\u663e\u5361\u3002<\/p>\n
\"\"<\/a><\/p>\n
 <\/p>\n
6.4 \u5b89\u88c5CUDA10.0<\/h3>\n
\u9009\u62e9CUDA\u7248\u672c\uff1a<\/strong><\/p>\n
CUDA10.2\u7248\u672c\u4e0eUbuntu\/GCC\u7248\u672c\u5bf9\u5e94\u5173\u7cfb<\/a><\/p>\n\n\n\n\n\n\n\n\n\n\n\n\n\n\n\n\n
Table 1. Native Linux Distribution Support in CUDA 10.2<\/span><\/span><\/span><\/caption>\n
Distribution<\/th>\nKernel*<\/th>\nGCC<\/th>\nGLIBC<\/th>\nICC<\/th>\nPGI<\/th>\nXLC<\/th>\nCLANG<\/th>\n<\/tr>\n<\/thead>\n
x86_64<\/td>\n<\/tr>\n
RHEL 8.1<\/td>\n4.18<\/td>\n8.2.1<\/td>\n2.28<\/td>\n<\/td>\n<\/td>\n<\/td>\n<\/td>\n<\/tr>\n
RHEL 7.7<\/td>\n3.10<\/td>\n4.8.5<\/td>\n2.17<\/td>\n19.0<\/td>\n18.x, 19.x<\/td>\nNO<\/td>\n8.0.0<\/td>\n<\/tr>\n
RHEL 6.10<\/td>\n2.6.32<\/td>\n4.4.7<\/td>\n2.12<\/td>\n<\/tr>\n
CentOS 7.7<\/td>\n3.10<\/td>\n4.8.5<\/td>\n2.17<\/td>\n<\/tr>\n
CentOS 6.10<\/td>\n2.6.32<\/td>\n4.4.7<\/td>\n2.12<\/td>\n<\/tr>\n
Fedora 29<\/td>\n4.16<\/td>\n8.0.1<\/td>\n2.27<\/td>\n<\/tr>\n
OpenSUSE Leap 15.1<\/td>\n4.15.0<\/td>\n7.3.1<\/td>\n2.26<\/td>\n<\/tr>\n
SLES 15.1<\/td>\n4.12.14<\/td>\n7.2.1<\/td>\n2.26<\/td>\n<\/tr>\n
SLES 12.4<\/td>\n4.12.14<\/td>\n4.8.5<\/td>\n2.22<\/td>\n<\/tr>\n
Ubuntu 18.04.3 (**)<\/span><\/span><\/td>\n4.15.0<\/td>\n7.3.0<\/span><\/td>\n2.27<\/td>\n<\/tr>\n
Ubuntu 16.04.6 (**)<\/td>\n4.4<\/td>\n5.4.0<\/td>\n2.23<\/td>\n<\/tr>\n<\/tbody>\n<\/table>\n
CUDA11.0\u7248\u672c\u4e0eUbuntu\/GCC\u7248\u672c\u5bf9\u5e94\u5173\u7cfb<\/a><\/p>\n\n\n\n\n\n\n\n\n\n\n\n\n\n
Table 1. Native Linux Distribution Support in CUDA 11.0<\/span><\/span><\/span><\/caption>\n
Distribution<\/th>\nKernel*<\/th>\nDefault GCC<\/th>\nGLIBC<\/th>\nGCC<\/th>\nICC<\/th>\nPGI<\/th>\nXLC<\/th>\nCLANG<\/th>\nArm C\/C++<\/th>\n<\/tr>\n<\/thead>\n
x86_64<\/td>\n<\/tr>\n
RHEL 8.1<\/td>\n4.18<\/td>\n8.3.1<\/td>\n2.28<\/td>\n9.x<\/span><\/td>\n19.1<\/td>\n19.x, 20.x<\/td>\nNO<\/td>\n9.0.0<\/td>\nNO<\/td>\n<\/tr>\n
CentOS 8.1<\/td>\n4.18<\/td>\n8.2.1<\/td>\n2.28<\/td>\n<\/tr>\n
RHEL 7.7<\/td>\n3.10<\/td>\n4.8.5<\/td>\n2.17<\/td>\n<\/tr>\n
CentOS 7.7<\/td>\n3.10<\/td>\n4.8.5<\/td>\n2.17<\/td>\n<\/tr>\n
OpenSUSE Leap 15.1<\/td>\n4.15.0<\/td>\n7.3.1<\/td>\n2.26<\/td>\n<\/tr>\n
SUSE SLES 15.1<\/td>\n4.12.14<\/td>\n7.2.1<\/td>\n2.26<\/td>\n<\/tr>\n
Ubuntu 18.04.4<\/span> (**)<\/td>\n4.15.0<\/td>\n7.4.0<\/span><\/td>\n2.27<\/td>\n<\/tr>\n
Ubuntu 16.04.6 (**)<\/td>\n4.4<\/td>\n5.4.0<\/td>\n2.23<\/td>\n<\/tr>\n<\/tbody>\n<\/table>\n
\u53ef\u4ee5\u770b\u5230CUDA11\u5728Ubuntu18.04\u4e0b\u9700\u8981GCC9.x\uff0cCUDA10.0\/10.2\u5728Ubuntu18.04\u4e0b\u9700\u8981GCC7.3\uff0cCUDA10.1\u5728Ubuntu18.04\u4e0b\u9700\u8981GCC7.4\uff0c\u800c\u6211\u521a\u5b89\u88c5\u7684Ubuntu18.04\u5b9e\u9645\u9ed8\u8ba4\u5e26\u7684\u662fGCC7.5\u7248\u672c\uff0c\u6240\u4ee5\u5e94\u8be5\u5b89\u88c5CUDA10\u3002<\/p>\n
Nvidia\u663e\u5361\u9a71\u52a8\u4e0eCUDA\u7248\u672c\u5bf9\u5e94\u5173\u7cfb<\/a><\/p>\n\n\n\n\n\n\n\n\n\n\n\n\n\n\n\n
Table 1. CUDA Toolkit and Compatible Driver Versions<\/span><\/caption>\n
CUDA Toolkit<\/th>\nLinux x86_64 Driver Version<\/th>\nWindows x86_64 Driver Version<\/th>\n<\/tr>\n<\/thead>\n
CUDA 10.2.89<\/span><\/td>\n>= 440.33<\/span><\/td>\n>= 441.22<\/td>\n<\/tr>\n
CUDA 10.1 (10.1.105 general release, and updates)<\/td>\n>= 418.39<\/td>\n>= 418.96<\/td>\n<\/tr>\n
CUDA 10.0.130<\/td>\n>= 410.48<\/td>\n>= 411.31<\/td>\n<\/tr>\n
CUDA 9.2 (9.2.148 Update 1)<\/td>\n>= 396.37<\/td>\n>= 398.26<\/td>\n<\/tr>\n
CUDA 9.2 (9.2.88)<\/td>\n>= 396.26<\/td>\n>= 397.44<\/td>\n<\/tr>\n
CUDA 9.1 (9.1.85)<\/td>\n>= 390.46<\/td>\n>= 391.29<\/td>\n<\/tr>\n
CUDA 9.0 (9.0.76)<\/td>\n>= 384.81<\/td>\n>= 385.54<\/td>\n<\/tr>\n
CUDA 8.0 (8.0.61 GA2)<\/td>\n>= 375.26<\/td>\n>= 376.51<\/td>\n<\/tr>\n
CUDA 8.0 (8.0.44)<\/td>\n>= 367.48<\/td>\n>= 369.30<\/td>\n<\/tr>\n
CUDA 7.5 (7.5.16)<\/td>\n>= 352.31<\/td>\n>= 353.66<\/td>\n<\/tr>\n
CUDA 7.0 (7.0.28)<\/td>\n>= 346.46<\/td>\n>= 347.62<\/td>\n<\/tr>\n<\/tbody>\n<\/table>\n
cat \/proc\/driver\/nvidia\/version\u00a0 \u53ef\u4ee5\u770b\u5230\u6211\u4eec\u5b89\u88c5\u7684\u9a71\u52a8\u662f44.82\uff0c\u6ee1\u8db3CUDA10.x\u7684\u9a71\u52a8\u4f9d\u8d56\u6761\u4ef6\u3002<\/p>\n
Tensorflow\u7248\u672c\u4e0eCUDA\/CUDNN \u5bf9\u5e94\u5173\u7cfb<\/a><\/p>\n
GPU<\/h4>\n
\n\n\n\n\n\n\n\n\n
\u7248\u672c<\/th>\nPython \u7248\u672c<\/th>\n\u7f16\u8bd1\u5668<\/th>\n\u6784\u5efa\u5de5\u5177<\/th>\ncuDNN<\/th>\nCUDA<\/th>\n<\/tr>\n
tensorflow-2.1.0<\/td>\n2.7\u30013.5-3.7<\/td>\nGCC 7.3.1<\/td>\nBazel 0.27.1<\/td>\n7.6<\/span><\/td>\n10.1<\/span><\/td>\n<\/tr>\n
tensorflow-2.0.0<\/td>\n2.7\u30013.3-3.7<\/td>\nGCC 7.3.1<\/td>\nBazel 0.26.1<\/td>\n7.4<\/span><\/td>\n10.0<\/span><\/td>\n<\/tr>\n
tensorflow_gpu-1.14<\/span>.0<\/td>\n2.7\u30013.3-3.7<\/td>\nGCC 4.8<\/td>\nBazel 0.24.1<\/td>\n7.4<\/span><\/td>\n10.0<\/span><\/td>\n<\/tr>\n
tensorflow_gpu-1.13.1<\/td>\n2.7\u30013.3-3.7<\/td>\nGCC 4.8<\/td>\nBazel 0.19.2<\/td>\n7.4<\/span><\/td>\n10.0<\/span><\/td>\n<\/tr>\n
tensorflow_gpu-1.12.0<\/td>\n2.7\u30013.3-3.6<\/td>\nGCC 4.8<\/td>\nBazel 0.15.0<\/td>\n7<\/td>\n9<\/td>\n<\/tr>\n<\/tbody>\n<\/table>\n<\/div>\n
PyTorch\u4e0eCUDA\u7248\u672c\u5bf9\u5e94\u5173\u7cfb<\/a><\/p>\n
pip3 install tensorflow-gpu\u00a0 \u9ed8\u8ba4\u5b89\u88c5\u4e86tensorflow-gpu-1.14.0\uff0cpytorch\u57fa\u672c1.2.0\u4ee5\u4e0a\u4e5f\u90fd\u53ef\u4ee5\u7528CUDA10.0\uff0c\u6240\u4ee5\u8fd9\u91cc\u4fdd\u9669\u70b9\u8fd8\u662f\u5b89\u88c5CUDA10.0+CUDNN7.4<\/span>\u3002<\/p>\n
\u5b89\u88c5CUDA10.0\uff1a<\/strong><\/p>\n
\u4ecehttps:\/\/developer.nvidia.com\/cuda-10.0-download-archive<\/a> \u4e0b\u8f7dcuda-10.0\uff0c\u9009\u62e9linux + x86_64 + ubuntu + 18.04 + runfile(local)\uff0c<\/p>\n
\"\"<\/a><\/p>\n
#\u4e0b\u8f7d\u5b89\u88c5cuda10.0    \uff08\u624b\u5de5\u5b89\u88c5\uff0c\u522b\u7528sudo apt install nvidia-cuda-toolkit\u88c5\uff0c\u7248\u672c\u4f1a\u5f88\u65e7\uff09\r\nwget https:\/\/developer.download.nvidia.cn\/compute\/cuda\/10.0\/secure\/Prod\/local_installers\/cuda_10.0.130_410.48_linux.run\r\nchmod +x cuda_10.0.130_410.48_linux.run\r\nsudo sh cuda_10.0.130_410.48_linux.run\r\n\r\n  Do you accept the previously read EULA?\r\n  accept\/decline\/quit: accept     \r\n\r\n  Install NVIDIA Accelerated Graphics Driver for Linux-x86_64 410.48?\r\n  (y)es\/(n)o\/(q)uit: n    #\u524d\u8fb9\u5df2\u7ecf\u88c5\u8fc7\u663e\u5361\u9a71\u52a8\u4e86\uff0c\u8fd9\u91cc\u4e0d\u8981\u91cd\u590d\u5b89\u88c5\r\n\r\n  Install the CUDA 10.0 Toolkit?\r\n  (y)es\/(n)o\/(q)uit: y\r\n\r\n  Enter Toolkit Location\r\n   [ default is \/usr\/local\/cuda-10.0 ]: \r\n\r\n  Do you want to install a symbolic link at \/usr\/local\/cuda?\r\n  (y)es\/(n)o\/(q)uit: y\r\n\r\n  Install the CUDA 10.0 Samples?\r\n  (y)es\/(n)o\/(q)uit: y\r\n\r\n  Enter CUDA Samples Location\r\n   [ default is \/home\/work ]: \r\n\r\n  Installing the CUDA Toolkit in \/usr\/local\/cuda-10.0 ...\r\n  Installing the CUDA Samples in \/home\/work ...\r\n  Copying samples to \/home\/work\/NVIDIA_CUDA-10.0_Samples now...\r\n  Finished copying samples.\r\n\r\n\r\n# \u4e0b\u8f7d\u5b89\u88c5cuda10.0 path1\u8865\u4e01\r\nwget https:\/\/developer.download.nvidia.cn\/compute\/cuda\/10.0\/Prod\/patches\/1\/cuda_10.0.130.1_linux.run\r\nchmod +x cuda_10.0.130.1_linux.run\r\nsudo sh cuda_10.0.130.1_linux.run<\/code><\/pre>\n
\"\"<\/a><\/h3>\n
# \u67e5\u770bcuda\u7248\u672c\r\ncd \/usr\/local\/cuda-10.0\/bin\r\nnvcc --version\r\n    nvcc: NVIDIA (R) Cuda compiler driver Copyright (c) 2005-2018 NVIDIA Corporation Built on Sat_Aug_25_21:08:01_CDT_2018 Cuda compilation tools, release 10.0, V10.0.130 \r\n\r\n# \u6dfb\u52a0\u73af\u5883\u53d8\u91cf\r\nsudo vim ~\/.bashrc\r\n    export PATH=\/usr\/local\/cuda-10.0\/bin${PATH:+:${PATH}}\r\n    export LD_LIBRARY_PATH=\/usr\/local\/cuda-10.0\/lib64${LD_LIBRARY_PATH:+:${LD_LIBRARY_PATH}}\r\n    export CUDA_HOME=\/usr\/local\/cuda-10.0\r\nsource ~\/.bashrc\r\n\r\n# \u6d4b\u8bd5cuda\r\ncd \/usr\/local\/cuda-10.0\/samples\/1_Utilities\/deviceQuery\r\nsudo make\r\nsudo .\/deviceQuery<\/code><\/pre>\n
\"\"<\/a><\/h3>\n
\"\"<\/a><\/h3>\n
\u663e\u793a PASS\uff0c\u8868\u793aCUDA\u5b89\u88c5\u6210\u529f\u3002<\/p>\n
\u6b64\u65f6\u987a\u4fbf\u518d\u662f\u4e0bnvidia-smi\uff0c\u770b\u770bgpu\u9a71\u52a8\u662f\u5426\u8fd8\u80fd\u6b63\u5e38\u5de5\u4f5c\u3002<\/p>\n
 <\/p>\n
6.5 \u5b89\u88c5cuDNN7.4<\/h3>\n
https:\/\/developer.nvidia.com\/rdp\/cudnn-archive<\/a><\/p>\n
\u6ce8\u518c\u8d26\u53f7\uff0c\u4e0b\u8f7d\u5bf9\u5e94cudnn\u7248\u672c\uff0c\u6211\u9009\u62e9\u7684\u9009\u62e9Download cuDNN v7.4.2 (Dec 14, 2018), for CUDA 10.0<\/a>\u4e0b\u7684cuDNN Library for Linux<\/a><\/p>\n
#\u4e0b\u8f7dCUDNN\r\nwget https:\/\/developer.download.nvidia.cn\/compute\/machine-learning\/cudnn\/secure\/v7.4.2\/prod\/10.0_20181213\/cudnn-10.0-linux-x64-v7.4.2.24.tgz\r\n\r\n#\u89e3\u538b\u540e\u7684\u6587\u4ef6\u5939\u540d\u79f0\u4e3acuda ,\u5c06\u5bf9\u5e94\u6587\u4ef6\u590d\u5236\u5230 \/usr\/local\u4e2d\u7684cuda-10.0\u76ee\u5f55\u5185\r\ntar zxvf cudnn-10.0-linux-x64-v7.4.2.24.tgz  \r\n# \u590d\u5236cudnn\u5934\u6587\u4ef6\r\nsudo cp .\/cuda\/include\/* \/usr\/local\/cuda-10.0\/include\/\r\n# \u590d\u5236cudnn\u7684\u5e93\r\nsudo cp .\/cuda\/lib64\/* \/usr\/local\/cuda-10.0\/lib64\/\r\n# \u6dfb\u52a0\u53ef\u6267\u884c\u6743\u9650\r\nsudo chmod +x \/usr\/local\/cuda-10.0\/include\/cudnn.h\r\nsudo chmod +x \/usr\/local\/cuda-10.0\/lib64\/libcudnn*\r\n\r\n#\u68c0\u67e5cudnn\u662f\u5426\u5b89\u88c5\u597d\r\ncat \/usr\/local\/cuda-10.0\/include\/cudnn.h | grep CUDNN_MAJOR -A 2\r\n<\/code><\/pre>\n
 <\/p>\n
7.\u5b89\u88c5pytorch\/tensorflow-gpu\uff0c\u6d4b\u8bd5cuda\u662f\u5426\u751f\u6548<\/h1>\n
sudo apt install python3-pip\r\npip3 install torch tensorflow-gpu\r\n\r\nvim test_gpu.py\r\n  import torch\r\n  print(torch.cuda.is_available())\r\n\r\n  device = torch.device(\"cuda:0\")\r\n  print(device)\r\n  print(torch.cuda.get_device_name(0))\r\n  print(torch.rand(3,3).cuda()) \r\n  print(torch.rand(5,5).cuda() + torch.rand(5,5).cuda())\r\n\r\n  import tensorflow as tf\r\n  print(tf.test.is_gpu_available())\r\n\r\npython test_gpu.py\r\n  True\r\n  cuda:0\r\n  GeForce RTX 2070\r\n  tensor([[0.9530, 0.4746, 0.9819],\r\n          [0.7192, 0.9427, 0.6768],\r\n          [0.8594, 0.9490, 0.6551]], device='cuda:0')\r\n  True\r\n\r\nnvidia-smi -l   #\u67e5\u770bgpu\u5360\u7528\u8fdb\u7a0b\u60c5\u51b5<\/code><\/pre>\n
 <\/p>\n
yan 20.6.13 0:55<\/p>\n
 <\/p>\n
\u53c2\u8003\uff1a<\/p>\n
\u53cc\u7cfb\u7edf\u5b89\u88c5Ubantu18.04 + RTX2070 + CUDA10.1 + Cudnn7.5.1+anaconda3+tensorflow-gpu<\/a><\/p>\n
win10\u4e0b\u5b89\u88c5Ubuntu16.04\u53cc\u7cfb\u7edf<\/a><\/p>\n
Linux\uff08Ubuntu\uff09\u6362apt-get\u6e90<\/a><\/p>\n
pip\/pip3\u66f4\u6362\u56fd\u5185\u955c\u50cf\u6e90<\/a><\/p>\n
Ubuntu18.04\u4e0b\u5b89\u88c5Pytorch-GPU<\/a><\/p>\n
Nvidia\u663e\u5361\u9a71\u52a8\u4e0eCUDA\u7248\u672c\u5bf9\u5e94\u5173\u7cfb<\/a><\/p>\n
CUDA\u7248\u672c\u4e0eUbuntu \u7248\u672c\uff0c\u4ee5\u53caGCC\u7248\u672c\u5bf9\u5e94\u5173\u7cfb<\/a><\/p>\n
Tensorflow\u7248\u672c\u4e0eCUDA\/CUDNN \u5bf9\u5e94\u5173\u7cfb<\/a><\/p>\n
 <\/p>\n","protected":false},"excerpt":{"rendered":"
\u7533\u8bf7\u4e86\u597d\u4e45\u7684\u9884\u7b97\u8001\u5a46\u7ec8\u4e8e\u6279\u4e86\uff0c\u5165\u624b\u534e\u7855\u7684GPU\u672c\uff08RTX2070\uff09\uff0c\u9ed8\u8ba4\u9884\u88c5\u4e86\u6b63\u7248win10\uff0c\u5e73\u65f6\u8fd8\u662f\u4e60\u60ef\u7528l […]<\/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,1001,855,1000],"_links":{"self":[{"href":"https:\/\/yanjingang.com\/blog\/index.php?rest_route=\/wp\/v2\/posts\/3864"}],"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=3864"}],"version-history":[{"count":3,"href":"https:\/\/yanjingang.com\/blog\/index.php?rest_route=\/wp\/v2\/posts\/3864\/revisions"}],"predecessor-version":[{"id":8785,"href":"https:\/\/yanjingang.com\/blog\/index.php?rest_route=\/wp\/v2\/posts\/3864\/revisions\/8785"}],"wp:attachment":[{"href":"https:\/\/yanjingang.com\/blog\/index.php?rest_route=%2Fwp%2Fv2%2Fmedia&parent=3864"}],"wp:term":[{"taxonomy":"category","embeddable":true,"href":"https:\/\/yanjingang.com\/blog\/index.php?rest_route=%2Fwp%2Fv2%2Fcategories&post=3864"},{"taxonomy":"post_tag","embeddable":true,"href":"https:\/\/yanjingang.com\/blog\/index.php?rest_route=%2Fwp%2Fv2%2Ftags&post=3864"}],"curies":[{"name":"wp","href":"https:\/\/api.w.org\/{rel}","templated":true}]}}