前言：
距離第一次安裝深度學(xué)習(xí)的GPU環(huán)境已經(jīng)過去了4年多（當(dāng)時(shí)TensorFlow特別麻煩），現(xiàn)在發(fā)現(xiàn)安裝pytorch的GPU版本還是很簡(jiǎn)單方便的，流程記錄如下。
安裝步驟：
步驟一：官網(wǎng)下載Anaconda
Free Download | Anaconda
直接下載最新版本到電腦里，并安裝。
步驟二：查詢電腦的CUDA Version
win+R 然后輸入cmd調(diào)出命令窗，輸入 nvidia-smi
步驟三：確定電腦GPU的NVDIA型號(hào)。
通過搜索找到”設(shè)備管理器”，再找到其中的“顯示適配器”。
?
步驟四：更新NVIDIA驅(qū)動(dòng)程序
NVIDIA官網(wǎng)：https://www.nvidia.cn/
?選擇對(duì)應(yīng)的硬件環(huán)境
?搜索到合適的驅(qū)動(dòng)后，下載并安裝（可以用C盤的默認(rèn)位置）。
?安裝完畢后，重啟電腦。
步驟五：再次查詢CUDA Version
win+R 然后輸入cmd調(diào)出命令窗，輸入 nvidia-smi
可以看到，CUDA Version的版本已經(jīng)從11.7變成了12.2。
步驟六：通過官網(wǎng)下載pytorch
官網(wǎng)地址：https://pytorch.org/
注意：CUDA Version要選擇比電腦低的情況，我這里選擇了默認(rèn)推薦版本。
在Anaconda的prompt中輸入官網(wǎng)指令。
?pip3 install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu117
在安裝過程中，遇到的選擇，全部選擇y。
其中，由于沒有借助鏡像源（增加出問題的概率），安裝過程較慢（大概花了30分鐘）。
?步驟七：在anaconda的prompt環(huán)境測(cè)試安裝結(jié)果
顯示True，安裝成功。
步驟八：打開anaconda自帶的spyder編輯器進(jìn)行測(cè)試
步驟九：跑一個(gè)MNIST數(shù)據(jù)集分類樣例代碼測(cè)試
?nvidia-smi -l 1實(shí)時(shí)觀察GPU的占用率
?補(bǔ)充說明
（1）windows系統(tǒng)需要增加Anaconda的環(huán)境變量。?
文章來(lái)源：http://www.zghlxwxcb.cn/news/detail-686235.html
（2）matplotlib通常需要重新安裝對(duì)應(yīng)版本，很容易版本不匹配。文章來(lái)源地址http://www.zghlxwxcb.cn/news/detail-686235.html
到了這里，關(guān)于基于全新電腦環(huán)境安裝pytorch的GPU版本的文章就介紹完了。如果您還想了解更多內(nèi)容，請(qǐng)?jiān)谟疑辖撬阉鱐OY模板網(wǎng)以前的文章或繼續(xù)瀏覽下面的相關(guān)文章，希望大家以后多多支持TOY模板網(wǎng)！
![Mac電腦配置李沐深度學(xué)習(xí)環(huán)境[pytorch版本]使用vscode](https://imgs.yssmx.com/Uploads/2024/02/742228-1.png)
