安裝環(huán)境

OS：win10專業(yè)版

Python：3.9.5

CUDA：11.7

cuDNN：cuDNN v8.4.1

?

參考資料：

[1]CUDA與CUDNN_五葉23333的博客-CSDN博客_cuda cudnn

[2]CUDA安裝教程（超詳細(xì)）_Billie使勁學(xué)的博客-CSDN博客_cuda安裝

[3]Win10 + CUDA11.7+pytorch手動(dòng)安裝-2.0版本_三世的博客-CSDN博客

安裝過(guò)程

首先了解一下CUDA和cuDNN

CUDA(ComputeUnified Device Architecture)，是顯卡廠商N(yùn)VIDIA推出的運(yùn)算平臺(tái)。 CUDA是一種由NVIDIA推出的通用并行計(jì)算架構(gòu)，該架構(gòu)使GPU能夠解決復(fù)雜的計(jì)算問(wèn)題。

NVIDIA cuDNN是用于深度神經(jīng)網(wǎng)絡(luò)的GPU加速庫(kù)。它強(qiáng)調(diào)性能、易用性和低內(nèi)存開(kāi)銷。NVIDIA cuDNN可以集成到更高級(jí)別的機(jī)器學(xué)習(xí)框架中，如谷歌的Tensorflow、加州大學(xué)伯克利分校的流行caffe軟件。簡(jiǎn)單的插入式設(shè)計(jì)可以讓開(kāi)發(fā)人員專注于設(shè)計(jì)和實(shí)現(xiàn)神經(jīng)網(wǎng)絡(luò)模型，而不是簡(jiǎn)單調(diào)整性能，同時(shí)還可以在GPU上實(shí)現(xiàn)高性能現(xiàn)代并行計(jì)算[1]。

如果你的電腦沒(méi)有nVidia的顯卡，那么就不用安裝cuda和cndnn，pytorch的版本也需要相應(yīng)的改變。?

1、查看你的顯卡對(duì)應(yīng)的CUDA版本

打開(kāi)NVIDIA控制面板，點(diǎn)擊左下角的系統(tǒng)信息，點(diǎn)擊組件，第三行的可以看出CUDA版本

2、下載安裝CUDA

CUDA官網(wǎng)：https://developer.nvidia.com/cuda-toolkit-archive

從中選擇符合的版本

安裝過(guò)程中注意記下安裝的路徑，后面有可能需要自己配置一下環(huán)境變量

具體的安裝過(guò)程可以看下面的鏈接

https://blog.csdn.net/m0_45447650/article/details/123704930?utm_medium=distribute.pc_feed_404.none-task-blog-2~default~BlogCommendFromBaidu~Rate-1-123704930-blog-null.pc_404_mixedpudn&depth_1-utm_source=distribute.pc_feed_404.none-task-blog-2~default~BlogCommendFromBaidu~Rate-1-123704930-blog-null.pc_404_mixedpud

3、安裝cuDNN

官網(wǎng)：https://developer.nvidia.com/rdp/cudnn-download

進(jìn)入官網(wǎng)，需要注冊(cè)一個(gè)賬號(hào)，登錄后點(diǎn)擊同意，從列表中找到符合你操作系統(tǒng)和CUDA版本要求的版本

下載的文件是一個(gè)壓縮包，將壓縮包解壓得到三個(gè)文件夾，復(fù)制到CUDA安裝目錄文件夾下，里面有和這三個(gè)文件夾同名的三個(gè)文件夾，將同名的文件夾中的內(nèi)容復(fù)制進(jìn)去，最后在路徑中添加如下路徑

安裝路徑\bin

安裝路徑\include

安裝路徑\lib

安裝路徑\libnvvp

其中安裝路徑是之前安裝CUDA的安裝路徑

具體操作方法可以看剛才鏈接的后半部分

https://blog.csdn.net/m0_45447650/article/details/123704930?utm_medium=distribute.pc_feed_404.none-task-blog-2~default~BlogCommendFromBaidu~Rate-1-123704930-blog-null.pc_404_mixedpudn&depth_1-utm_source=distribute.pc_feed_404.none-task-blog-2~default~BlogCommendFromBaidu~Rate-1-123704930-blog-null.pc_404_mixedpud

4、安裝pytorch

pytorch官網(wǎng)：https://pytorch.org/get-started/locally/

在這里點(diǎn)擊選擇你的安裝環(huán)境，官網(wǎng)會(huì)給你自動(dòng)生成一條命令，在cmd中執(zhí)行這條命令就可以自動(dòng)安裝pytorch

? ?? ? ? https://blog.csdn.net/qimo601/article/details/125146818

上面這個(gè)鏈接講了如何手動(dòng)下載庫(kù)安裝包并本地安裝，下載速度可以更快，但我實(shí)測(cè)發(fā)現(xiàn)直接用pip install下載速度并不慢，維持在6~7M每秒，完全可以接受，如果你的網(wǎng)絡(luò)情況不佳，可以試試這個(gè)方法。

5、安裝完成

在cmd中輸入python，打開(kāi)python。

導(dǎo)入torch包測(cè)試是否安裝成功? ?

? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ??? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ??