{"id":386,"date":"2024-03-13T08:44:06","date_gmt":"2024-03-13T00:44:06","guid":{"rendered":"http:\/\/www.ccwifi.cc\/blogs\/?p=386"},"modified":"2024-03-13T15:52:25","modified_gmt":"2024-03-13T07:52:25","slug":"linux%e5%ae%89%e8%a3%85nvidia%e6%98%be%e5%8d%a1%e9%a9%b1%e5%8a%a8-cuda-cudnn-pytorch","status":"publish","type":"post","link":"https:\/\/www.ccwifi.cc\/blogs\/2024\/03\/13\/linux%e5%ae%89%e8%a3%85nvidia%e6%98%be%e5%8d%a1%e9%a9%b1%e5%8a%a8-cuda-cudnn-pytorch\/","title":{"rendered":"Linux\u5b89\u88c5Nvidia\u663e\u5361\u9a71\u52a8+CUDA+cuDNN+PyTorch"},"content":{"rendered":"
\n

\u5728\u7ebfwifi\u8dd1\u5305 \u91d1\u521a\u5305\u8dd1\u5305 cap\u8dd1\u5305 hccapx ewsa\u5728\u7ebf \u5c31\u6765 \u66f9\u64cdwifi<\/a><\/strong><\/p>\n

\u5404\u4f4d\u597d \u53c8\u89c1\u9762\u4e86 \u6211\u662f\u66f9\u64cd \u4eca\u5929\u7ed9\u5927\u5bb6\u5e26\u6765\u4e00\u7bc7\u65b0\u7684\u6559\u7a0b<\/p>\n

\u5e0c\u671b\u5404\u4f4d\u7ec6\u5fc3\u5b66\u4e60 \u4f4e\u8c03\u7528\u7f51<\/p>\n<\/div>\n

lspci |<\/span> grep -i nvidia<\/code><\/pre>\n
(<\/span>base)<\/span> root@pc:~# lspci |<\/span> grep -i nvidia\n01:00.0 VGA compatible controller: NVIDIA Corporation GP102 [<\/span>GeForce GTX 1080<\/span> Ti]<\/span> (<\/span>rev a1)<\/span>\n01:00.1 Audio device: NVIDIA Corporation GP102 HDMI Audio Controller (<\/span>rev a1)<\/span><\/code><\/pre>\n
\u6b64\u8fc7\u7a0b\u5751\u6bd4\u8f83\u591a\uff0c\u8bf7\u8c28\u614e\u64cd\u4f5c\uff0c\u6bcf\u4e00\u6b65\u90fd\u5f88\u91cd\u8981\uff0c\u4e0d\u8981\u8df3\u6b65\uff01\u6587\u7ae0\u8f83\u957f\uff0c\u8bf7\u8010\u5fc3\u9605\u8bfb\uff01\u672c\u6587\u6240\u4f7f\u7528\u7684Linux\u7cfb\u7edf\u4e3aUbuntu 18.04\u3002\u64cd\u4f5c\u6d41\u7a0b\u5982\u4e0b\uff1a\u67e5\u770b\u7535\u8111\u7684\u663e\u5361\u578b\u53f7\uff1a\u663e\u793a\u5982\u4e0b\uff1a<\/p>\n
sudo -s\n\u7136\u540e\u8f93\u5165\u5bc6\u7801\uff0c\u56de\u8f66<\/code><\/pre>\n
vim \/etc\/modprobe.d\/blacklist-nouveau.conf<\/code><\/pre>\n
\u53ef\u4ee5\u770b\u5230\u6211\u7684\u7535\u8111\u7684\u663e\u5361\u578b\u53f7\u4e3a\uff1aGeForce GTX 1080 Ti\u3002\u77e5\u9053\u663e\u5361\u7248\u672c\u4e4b\u540e\u5c31\u53ef\u4ee5\u5f00\u59cb\u5b89\u88c5\u663e\u5361\u9a71\u52a8\u4e86\u3002\u9996\u5148\u8fdb\u5165\u7ba1\u7406\u5458\u6743\u9650\uff1a\u5b89\u88c5Nvidia\u663e\u5361\u9a71\u52a8\u7981\u7528Nouveau\u9a71\u52a8\u3002\u9996\u5148\u8981\u7981\u7528ubuntu\u81ea\u5e26\u7684Nouveau\u9a71\u52a8\uff0cNouveau\u662f\u4e00\u4e2a\u7b2c\u4e09\u65b9\u5f00\u6e90\u9a71\u52a8\uff0c\u6211\u4eec\u6700\u597d\u4f7f\u7528nvidia\u5b98\u65b9\u7684\u9a71\u52a8\u3002\u7f16\u8f91\u6587\u4ef6\uff1a<\/p>\n
blacklist nouveau\noptions nouveau modeset<\/span>=<\/span>0<\/span><\/code><\/pre>\n
update-initramfs -u<\/code><\/pre>\n
reboot<\/code><\/pre>\n
lspci |<\/span> grep nouveau<\/code><\/pre>\n
apt-get install build-essential \napt-get install cmake<\/code><\/pre>\n
gcc --version\ncmake --version<\/code><\/pre>\n
\"hashcat\u5982\u4f55\u5b89\u88c5\u663e\u5361\u9a71\u52a8\"<\/p>\n
\u5728\u6587\u4ef6\u6700\u540e\u52a0\u5165\u4ee5\u4e0b\u5185\u5bb9\u3002\u66f4\u65b0\u4f7f\u5176\u751f\u6548\u3002\u91cd\u542f\u673a\u5668\u3002\u68c0\u67e5\u662f\u5426\u7981\u7528\u6210\u529f\u3002\u6ca1\u6709\u5185\u5bb9\u8f93\u51fa\uff0c\u8bf4\u660e\u7981\u7528\u6210\u529f\u3002\u5b89\u88c5gcc\/cmake\u3002\u68c0\u67e5\u662f\u5426\u5b89\u88c5\u6210\u529f\uff1a\u6709\u7248\u672c\u53f7\u663e\u793a\u8bf4\u660e\u5b89\u88c5\u6210\u529f\u3002\u5b89\u88c5NVIDIA\u9a71\u52a8\u4e0b\u8f7d\u9a71\u52a8\u3002\u5b98\u7f51\u4e0b\u8f7d\u9a71\u52a8\uff1a\u8fdb\u5165\u7f51\u7ad9\u9009\u62e9\u81ea\u5df1\u7684\u663e\u5361\u578b\u53f7\u3002<\/p>\n
\"hashcat\u5982\u4f55\u5b89\u88c5\u663e\u5361\u9a71\u52a8\"<\/p>\n
service lightdm stop<\/code><\/pre>\n
\u70b9\u51fb\u641c\u7d22\u4e4b\u540e\u4e0b\u65b9\u4f1a\u663e\u793a\u53ef\u4ee5\u4f7f\u7528\u7684\u9a71\u52a8\u3002\u9009\u62e9\u60f3\u8981\u4e0b\u8f7d\u7684\u9a71\u52a8\u7248\u672c\uff0c\u5e76\u70b9\u51fb\u201c\u83b7\u53d6\u4e0b\u8f7d\u201d\u8fdb\u884c\u4e0b\u8f7d\u3002\u8fd9\u4e9b\u7248\u672c\u90fd\u53ef\u4ee5\u4f7f\u7528\uff0c\u53ea\u4e0d\u8fc7\u4e0d\u540c\u7248\u672c\u5bf9\u5e94\u7684cuda\u7248\u672c\u4e0d\u4e00\u6837\uff0c\u5bf9\u5e94\u7684cudnn\u7248\u672c\u4e5f\u5c31\u4e0d\u4e00\u6837\uff0c\u6700\u540e\u652f\u6301\u7684pytorch\u7248\u672c\u4e5f\u5c31\u4e0d\u4e00\u6837\uff0c\u6240\u4ee5\uff0c\u9009\u62e9\u54ea\u4e2a\u7248\u672c\u90fd\u884c\uff0c\u53ea\u4e0d\u8fc7\u540e\u7eed\u5b89\u88c5cuda\u3001cudnn\u3001pytorch\u7684\u65f6\u5019\u9700\u8981\u8fdb\u884c\u7248\u672c\u5bf9\u5e94\u3002\u4e0b\u8f7d\u5b8c\u6210\u4e4b\u540e\u8bb0\u4f4f\u6587\u4ef6\u4e0b\u8f7d\u5230\u7684\u76ee\u5f55\u3002\u5173\u95ed\u663e\u793a\u6a21\u5f0f\u3002\u540c\u65f6\u6309\u4f4fCtrl+Alt+F3\u8fdb\u5165tty\u7ec8\u7aef\u6a21\u5f0f\u3002\u8fdb\u5165\u7ec8\u7aef\u6a21\u5f0f\u4e4b\u540e\u9700\u8981\u518d\u6b21\u8f93\u5165\u8ba1\u7b97\u673a\u7684\u7528\u6237\u540d\u548c\u5bc6\u7801\u3002\u8f93\u5165\u4e0b\u9762\u547d\u4ee4\u5173\u95ed\u663e\u793a\u7ba1\u7406\uff1a<\/p>\n
cd<\/span> \/home\/username\/Downloads  # \u6b64\u5904\u7684\u76ee\u5f55\u4e3a\u4f60\u4e0b\u8f7d\u7684\u9a71\u52a8\u6587\u4ef6\u6240\u5728\u76ee\u5f55<\/span>\nsh NVIDIA-Linux-x86_64-415.27.run --no-opengl-files<\/code><\/pre>\n
\"hashcat\u5982\u4f55\u5b89\u88c5\u663e\u5361\u9a71\u52a8\"\"hashcat\u5982\u4f55\u5b89\u88c5\u663e\u5361\u9a71\u52a8\"\"hashcat\u5982\u4f55\u5b89\u88c5\u663e\u5361\u9a71\u52a8\"\"hashcat\u5982\u4f55\u5b89\u88c5\u663e\u5361\u9a71\u52a8\"<\/p>\n
(<\/span>base)<\/span> root@pc:~# nvidia-smi\nThu Apr 22<\/span> 10:19:06 2021<\/span>       \n+-----------------------------------------------------------------------------+\n|<\/span> NVIDIA-SMI 415.27       Driver Version: 415.27       CUDA Version: 10.0     |<\/span>\n|<\/span>-------------------------------+----------------------+----------------------+\n|<\/span> GPU  Name        Persistence-M|<\/span> Bus-Id        Disp.A |<\/span> Volatile Uncorr. ECC |<\/span>\n|<\/span> Fan  Temp  Perf  Pwr:Usage\/Cap|<\/span>         Memory-Usage |<\/span> GPU-Util  Compute M. |<\/span>\n|<\/span>===============================<\/span>+======================<\/span>+======================<\/span>|<\/span>\n|<\/span>   0<\/span>  GeForce GTX 108...  Off  |<\/span> 00000000:01:00.0  On |<\/span>                  N\/A |<\/span>\n|<\/span> 26%   60C    P8    17W \/ 280W |<\/span>    445MiB \/ 11177MiB |<\/span>      0%      Default |<\/span>\n+-------------------------------+----------------------+----------------------+\n                                                                               \n+-----------------------------------------------------------------------------+\n|<\/span> Processes:                                                       GPU Memory |<\/span>\n|<\/span>  GPU       PID   Type   Process name                             Usage      |<\/span>\n|<\/span>=============================================================================<\/span>|<\/span>\n|<\/span>    0<\/span>      1218<\/span>      G   \/usr\/lib\/xorg\/Xorg                            24MiB |<\/span>\n|<\/span>    0<\/span>      1271<\/span>      G   \/usr\/bin\/gnome-shell                          49MiB |<\/span>\n|<\/span>    0<\/span>      1460<\/span>      G   \/usr\/lib\/xorg\/Xorg                           159MiB |<\/span>\n|<\/span>    0<\/span>      1617<\/span>      G   \/usr\/bin\/gnome-shell                         152MiB |<\/span>\n|<\/span>    0<\/span>      1920<\/span>      G   fcitx-qimpanel                                49MiB |<\/span>\n|<\/span>    0<\/span>      5985<\/span>      G   \/usr\/lib\/firefox\/firefox                       2MiB |<\/span>\n|<\/span>    0<\/span>      6098<\/span>      G   \/usr\/lib\/firefox\/firefox                       2MiB |<\/span>\n+-----------------------------------------------------------------------------+\n(<\/span>base)<\/span> root@pc:~#<\/code><\/pre>\n
\u5b89\u88c5\u9a71\u52a8\u3002\u8fdb\u5165\u521a\u521a\u4e0b\u8f7d\u7684\u9a71\u52a8\u6587\u4ef6\u6240\u5728\u76ee\u5f55\uff0c\u5728\u547d\u4ee4\u884c\u7ec8\u7aef\u8fd0\u884c\u521a\u624d\u4e0b\u8f7d\u7684\u9a71\u52a8\u3002\u5b89\u88c5\u8fc7\u7a0b\u6309\u7167\u4e0b\u9762\u6b65\u9aa4\u8fdb\u884c\u9009\u62e9\u3002\u603b\u4e4b\u4e00\u8defcontinue\uff0c\u4e00\u8defok\u5c31\u884c\u4e86\u3002\u68c0\u67e5\u5b89\u88c5\u7ed3\u679c\u3002\u91cd\u542f\u673a\u5668\u540e\uff0c\u8f93\u5165nvidia-smi\u67e5\u770b\u8f93\u51fa\u663e\u793a\u7c7b\u4f3c\u4e0b\u9762\uff0c\u5c31\u8bf4\u660e\u5df2\u7ecf\u5b89\u88c5\u6210\u529f\u3002<\/p>\n
\"hashcat\u5982\u4f55\u5b89\u88c5\u663e\u5361\u9a71\u52a8\"\"hashcat\u5982\u4f55\u5b89\u88c5\u663e\u5361\u9a71\u52a8\"<\/p>\n
\u53ef\u4ee5\u770b\u5230\u5339\u914d\u7684CUDA\u7248\u672c\u4e3a10.0\u3002\u5b89\u88c5CUDA\u3002\u67e5\u770bNvidia\u663e\u5361\u9a71\u52a8\u5bf9\u5e94\u7684CUDA\u7248\u672c\u3002\u53ef\u67e5\u770b\u5b98\u7f51\u7ed9\u51fa\u7684\u5339\u914d\u8868\u3002\u53ef\u4ee5\u770b\u5230\u6211\u5b89\u88c5\u7684415.27\u7248\u672c\u7684\u9a71\u52a8\u5bf9\u5e94\u7684cuda\u7248\u672c\uff0cCUDA10.0\u53ca10.0\u4ee5\u4e0b\u7684\u7248\u672c\u90fd\u53ef\u4ee5\u7528\uff0c\u56e0\u4e3aCUDA10.1\u9700\u8981\u9a71\u52a8\u7248\u672c\u81f3\u5c11\u4e3a418.39\u3002\u4e0b\u8f7dCUDA\u3002\u4e0b\u8f7dCUDA:<\/p>\n
\"hashcat\u5982\u4f55\u5b89\u88c5\u663e\u5361\u9a71\u52a8\"<\/p>\n
sh cuda_10.0.130_410.48_linux.run --override<\/code><\/pre>\n
1<\/span> \u8f93\u5165 accept \u63a5\u53d7\u6761\u6b3e\n2<\/span> \u8f93\u5165n\u4e0d\u5b89\u88c5nvidia\u56fe\u50cf\u9a71\u52a8\uff0c\u4e4b\u524d\u5df2\u7ecf\u5b89\u88c5\u8fc7\u4e86 \/\/\u6b64\u5904\u4e00\u5b9a\u8981\u9009\u62e9n\uff01\uff01\uff01\n3<\/span> \u8f93\u5165y\u5b89\u88c5cuda 10.0\u5de5\u5177\n4<\/span> \u56de\u8f66\u786e\u8ba4cuda\u9ed8\u8ba4\u5b89\u88c5\u8def\u5f84\uff1a\/usr\/local\/cuda-10.0\n5<\/span> \u8f93\u5165y\u6216\u8005n\u5b89\u88c5\u6216\u8005\u4e0d\u5b89\u88c5\u6307\u5411\/usr\/local\/cuda\u7684\u7b26\u53f7\u94fe\u63a5\n6<\/span> \u8f93\u5165y\u5b89\u88c5CUDA 10.0 Samples\uff0c\u4ee5\u4fbf\u540e\u9762\u6d4b\u8bd5\n7<\/span> \u56de\u8f66\u786e\u8ba4CUDA 10.0 Samples\u9ed8\u8ba4\u5b89\u88c5\u8def\u5f84\uff1a<\/code><\/pre>\n
vim ~\/.bashrc  \n\/\/\u672b\u5c3e\u6dfb\u52a0\nexport<\/span> CUDA_HOME<\/span>=<\/span>\/usr\/local\/cuda-10.0    \nexport<\/span> LD_LIBRARY_PATH<\/span>=<\/span>\/usr\/local\/cuda-10.0\/lib64:$LD_LIBRARY_PATH<\/span>   \nexport<\/span> PATH<\/span>=<\/span>\/usr\/local\/cuda-10.0\/bin:$PATH<\/span><\/code><\/pre>\n
\u627e\u5230CUDA 10.0\u3002\u5b89\u88c5\u3002\u4e0b\u8f7d\u4e4b\u540e\u8fdb\u5165Terminal\uff0ccd\u8fdb\u5165\u6587\u4ef6\u6240\u5728\u76ee\u5f55\uff0c\u8f93\u5165\u3002\u5b89\u88c5\u8fc7\u7a0b\u4e2d\u4f1a\u51fa\u73b0\u5f88\u591a\u9009\u9879\uff0c\u6309\u4f4f\u2018S\u2019\u6ed1\u5230100%\u540e\uff0c\u6839\u636e\u63d0\u793a\u8f93\u5165\u3002\u6b64\u5904\u4e0d\u8981\u518d\u9009\u62e9\u5b89\u88c5\u9a71\u52a8\uff01\u6b64\u5904\u4e0d\u8981\u518d\u9009\u62e9\u5b89\u88c5\u9a71\u52a8\uff01\u6b64\u5904\u4e0d\u8981\u518d\u9009\u62e9\u5b89\u88c5\u9a71\u52a8\uff01\uff08\u867d\u7136\u6211\u4e5f\u4e0d\u6e05\u695a\u88c5\u4e86\u4f1a\u600e\u4e48\u6837\uff0c\u6709\u5174\u8da3\u7684\u5c0f\u4f19\u4f34\u53ef\u4ee5\u8bd5\u4e00\u8bd5\uff09\u3002\u5b89\u88c5\u5b8c\u6210\u540e\uff0c\u4f1a\u63d0\u51fa\u4e0d\u5b8c\u6574\u7684\u5b89\u88c5\uff0c\u6b64\u5904\u4e0d\u7528\u7ba1\uff0c\u662f\u56e0\u4e3a\u4e0a\u4e00\u6b65\u6ca1\u6709\u9009\u62e9\u5b89\u88c5\u663e\u5361\u9a71\u52a8\u5bfc\u81f4\u7684\u3002\u914d\u7f6e\u73af\u5883\u53d8\u91cf\u3002\u4e0b\u4e00\u6b65\u6dfb\u52a0\u73af\u5883\u8def\u5f84\u3002<\/p>\n
cd<\/span> \/usr\/local\/cuda-9.0\/samples\/1_Utilities\/deviceQuery  \nmake \nsudo .\/deviceQuery<\/code><\/pre>\n
\"hashcat\u5982\u4f55\u5b89\u88c5\u663e\u5361\u9a71\u52a8\"\"hashcat\u5982\u4f55\u5b89\u88c5\u663e\u5361\u9a71\u52a8\"\"hashcat\u5982\u4f55\u5b89\u88c5\u663e\u5361\u9a71\u52a8\"<\/p>\n
\u5230\u6b64\u5df2\u7ecf\u5b8c\u6210\u5b89\u88c5\uff0c\u4e0b\u9762\u6d4b\u8bd5\u5b89\u88c5\u662f\u5426\u6210\u529f\u3002\u5982\u679c\u8f93\u51fa\u7c7b\u4f3c\u4e8e\u4e0b\u56fe\u7684\u4e00\u5806\u4e1c\u897f\uff08\u56fe\u7247\u622a\u81ea\u7f51\u7edc\uff09\uff0c\u6700\u540e\u8fd4\u56deresult==PASS\u8bf4\u660eCUDA\u5b89\u88c5\u914d\u7f6e\u6210\u529f\u3002\u5b89\u88c5cuDNN\u3002\u4e0b\u8f7d\u5730\u5740\u3002\u53ef\u4ee5\u770b\u5230\u6bcf\u4e2acuDNN\u7248\u672c\u90fd\u6709\u5176\u5bf9\u5e94\u7684CUDA\u7248\u672c\uff0c\u6211\u7684CUDA\u4e3a10.0\uff0c\u6240\u4ee5\u627e\u5230\u5bf9\u5e94CUDA10.0\u7684\u4efb\u610fcuDNN\u7248\u672c\u4e0b\u8f7d\u5747\u53ef\u3002\u6211\u4e0b\u8f7d\u7684cuDNN v7.5.0 (Feb 21, 2019), for CUDA 10.0\u3002<\/p>\n
tar -zxvf cudnn-10.0-linux-x64-v7.5.0.56.tgz<\/code><\/pre>\n
\u9009\u62e9cuDNN Library for Linux\u4e0b\u8f7d\u5373\u53ef\uff0c\u4e0b\u8f7d\u4f1a\u63d0\u793a\u9700\u8981\u5148\u767b\u5f55\u4e00\u4e0b\u4ed6\u4eec\u7684\u7f51\u7ad9\uff0c\u767b\u5f55\u540e\u4e0b\u8f7d\u5373\u53ef\u3002\u4e0b\u8f7d\u5b8c\u6210\u4f1a\u5f97\u5230\u4e00\u4e2atgz\u538b\u7f29\u5305\uff1acudnn-10.0-linux-x64-v7.5.0.56.tgz\u3002\u8fdb\u5165\u4e0b\u8f7d\u7684\u6587\u4ef6\u6240\u5728\u76ee\u5f55\uff0c\u89e3\u538b\u3002<\/p>\n
cp cuda\/include\/cudnn.h  \/usr\/local\/cuda\/include \ncp cuda\/lib64\/libcudnn*  \/usr\/local\/cuda\/lib64<\/code><\/pre>\n
\"hashcat\u5982\u4f55\u5b89\u88c5\u663e\u5361\u9a71\u52a8\"<\/p>\n
\u89e3\u538b\u540e\u4f1a\u5f97\u5230\u4e00\u4e2acuda\u6587\u4ef6\u5939\u3002\u9700\u8981\u5c06cuda\/include\/cudnn.h\u548ccuda\/lib64\/libcudnn*\u62f7\u8d1d\u5230\u4e4b\u524d\u5b89\u88c5\u7684cuda\u7684\u6587\u4ef6\u5939\u4e0b\u3002\u81f3\u6b64\uff0ccuDNN\u5b89\u88c5\u5b8c\u6210\uff0c\u63a5\u4e0b\u6765\u5c31\u53ef\u4ee5\u5b89\u88c5PyTorch\u53bb\u8c03\u7528\u663e\u5361\u4e86\u3002\u5b89\u88c5PyTorch\u3002\u8fdb\u5165\u7f51\u7ad9\u627e\u5230\u5bf9\u5e94\u7248\u672c\u7684PyTorch\u3002<\/p>\n
\"hashcat\u5982\u4f55\u5b89\u88c5\u663e\u5361\u9a71\u52a8\"<\/p>\n
# pip:<\/span>\npip install torch<\/span>==<\/span>1.2.0 torchvision<\/span>==<\/span>0.4.0 -f  -i \n# conda:<\/span>\nconda install pytorch<\/span>==<\/span>1.2.0 torchvision<\/span>==<\/span>0.4.0 cudatoolkit<\/span>=<\/span>10.0 -c <\/code><\/pre>\n
\u4e00\u76f4\u5f80\u4e0b\u6ed1\uff0c\u53ef\u4ee5\u627e\u5230\u5bf9\u5e94CUDA10.0\u7684torch\u7248\u672c\uff0c\u53ef\u4ee5\u4f7f\u7528conda\u5b89\u88c5\uff0c\u4e5f\u53ef\u4ee5\u4f7f\u7528pip\u5b89\u88c5\uff0c\u5982\u679c\u7eaf\u7528\u7f51\u7ad9\u63d0\u4f9b\u7684\u6307\u4ee4\u4f1a\u4e0b\u8f7d\u5f88\u6162\uff0c\u53ef\u4ee5\u6362\u8c46\u74e3\u7684\u6e90\u6765\u5b89\u88c5\uff0c\u63a8\u8350\u4f7f\u7528\u4e0b\u65b9pip\u6307\u4ee4\u5b89\u88c5\uff08\u5df2\u9a8c\u8bc1\u6210\u529f\u5b89\u88c5\uff09\u3002<\/p>\n
# \u7ec8\u7aef\u8fdb\u5165python\uff1a<\/span>\npython<\/code><\/pre>\n
>>> import torch\n>>> torch.cuda.is_available()<\/span><\/code><\/pre>\n
<\/p>\n
\u5230\u6b64\u4e3a\u6b62\uff0c\u6240\u6709\u5b89\u88c5\u90fd\u5df2\u5b8c\u6210\uff01\u8c03\u7528\u663e\u5361\u3002\u4e0b\u9762\u6765\u68c0\u6d4btorch\u662f\u5426\u53ef\u4ee5\u8c03\u7528\u663e\u5361\u3002\u8c03\u7528\u663e\u5361\u3002\u5982\u679c\u8fd4\u56deTrue\uff0c\u606d\u559c\u4f60\uff0c\u6210\u529f\u4e86\uff01\u5982\u679c\u8fd4\u56deFalse\uff0c\u8bf7\u4ece\u5934\u68c0\u67e5\u5404\u4e2a\u6b65\u9aa4\u662f\u5426\u51fa\u73b0\u95ee\u9898\u3002Reference\uff1a1.blog.csdn.net\/weixin_40702638\/article\/details\/83304415 2.developer.nvidia.com\/cuda-toolkit-archive 3.pytorch.org\/get-started\/previous-versions\/ 4.docs.nvidia.com\/cuda\/cuda-toolkit-release-notes\/index.html 5.blog.csdn.net\/wohu1104\/article\/details\/107032493 6.nvidia.cn\/geforce\/drivers\/ 7.developer.nvidia.com\/rdp\/cudnn-archive
\n“`<\/p>\n","protected":false},"excerpt":{"rendered":"
Linux\u5b89\u88c5Nvidia\u663e\u5361\u9a71\u52a8+CUDA+cuDNN+PyTorch\u6b64\u8fc7\u7a0b\u5751\u6bd4\u8f83\u591a\uff0c\u8bf7\u8c28\u614e\u64cd\u4f5c\uff0c\u6bcf\u4e00\u6b65\u90fd\u5f88\u91cd\u8981\uff0c\u4e0d\u8981\u8df3\u6b65\uff01\u6587\u7ae0\u8f83\u957f\uff0c\u8bf7\u8010\u5fc3\u9605\u8bfb\uff01\u672c\u6587\u6240\u4f7f\u7528\u7684Linux\u7cfb\u7edf\u4e3a  Ubuntu 18.04\u3002<\/p>\n","protected":false},"author":1,"featured_media":387,"comment_status":"open","ping_status":"open","sticky":false,"template":"","format":"standard","meta":{"footnotes":""},"categories":[1],"tags":[],"topic":[],"class_list":["post-386","post","type-post","status-publish","format-standard","has-post-thumbnail","hentry","category-1"],"_links":{"self":[{"href":"https:\/\/www.ccwifi.cc\/blogs\/wp-json\/wp\/v2\/posts\/386","targetHints":{"allow":["GET"]}}],"collection":[{"href":"https:\/\/www.ccwifi.cc\/blogs\/wp-json\/wp\/v2\/posts"}],"about":[{"href":"https:\/\/www.ccwifi.cc\/blogs\/wp-json\/wp\/v2\/types\/post"}],"author":[{"embeddable":true,"href":"https:\/\/www.ccwifi.cc\/blogs\/wp-json\/wp\/v2\/users\/1"}],"replies":[{"embeddable":true,"href":"https:\/\/www.ccwifi.cc\/blogs\/wp-json\/wp\/v2\/comments?post=386"}],"version-history":[{"count":1,"href":"https:\/\/www.ccwifi.cc\/blogs\/wp-json\/wp\/v2\/posts\/386\/revisions"}],"predecessor-version":[{"id":420,"href":"https:\/\/www.ccwifi.cc\/blogs\/wp-json\/wp\/v2\/posts\/386\/revisions\/420"}],"wp:featuredmedia":[{"embeddable":true,"href":"https:\/\/www.ccwifi.cc\/blogs\/wp-json\/wp\/v2\/media\/387"}],"wp:attachment":[{"href":"https:\/\/www.ccwifi.cc\/blogs\/wp-json\/wp\/v2\/media?parent=386"}],"wp:term":[{"taxonomy":"category","embeddable":true,"href":"https:\/\/www.ccwifi.cc\/blogs\/wp-json\/wp\/v2\/categories?post=386"},{"taxonomy":"post_tag","embeddable":true,"href":"https:\/\/www.ccwifi.cc\/blogs\/wp-json\/wp\/v2\/tags?post=386"},{"taxonomy":"topic","embeddable":true,"href":"https:\/\/www.ccwifi.cc\/blogs\/wp-json\/wp\/v2\/topic?post=386"}],"curies":[{"name":"wp","href":"https:\/\/api.w.org\/{rel}","templated":true}]}}