1.確定操作系統(tǒng)及cuda版本

• 前置知識(shí)： 安裝好annaconda或者miniconda進(jìn)行python虛擬環(huán)境管理，建議miniconda。（好處是你可以在一臺(tái)主機(jī)上安裝多個(gè)互不影響的python虛擬環(huán)境，然后在運(yùn)行項(xiàng)目1時(shí)激活其對(duì)應(yīng)的python虛擬環(huán)境1，在運(yùn)行項(xiàng)目2時(shí)激活其對(duì)應(yīng)的python虛擬環(huán)境2）。當(dāng)然如果你是100%小白，先不管這些也問題不大，但我強(qiáng)烈建議你花點(diǎn)時(shí)間學(xué)一下，畢竟大概最多一個(gè)月后你還是要學(xué)的。
• 確認(rèn)操作系統(tǒng)： 這個(gè)很簡(jiǎn)單，主要是三類，windows、linux（Ubuntu、centos等）、macos。不同版本的系統(tǒng)底層gcc版本可能不太一樣，可能會(huì)影響下一步中的兼容性問題，不過一般不會(huì)遇到。
• 確認(rèn)系統(tǒng)中安裝的cuda版本： 在終端輸入 nvidia-smi,回車，例如下圖，版本號(hào)為11.8：
  

2.確定python pytorch cuda之間的版本是否兼容

在正式配置環(huán)境之前，需要檢查兼容性。（后續(xù)當(dāng)你對(duì)需要配置環(huán)境的主機(jī)或者服務(wù)器熟悉后，該步驟可省略）

• 對(duì)于python和pytorch，我沒有找到兼容性表格，但總體來說，舊的python對(duì)應(yīng)舊一點(diǎn)的pytorch，新的對(duì)應(yīng)新一點(diǎn)的基本都沒問題，python3.7-3.10，pytorch1.10-2.0。

• 然后最重要的是 pytorch和系統(tǒng)cuda之間的版本兼容問題。總體上來說你安裝的pytorch對(duì)應(yīng)的cuda版本要小于等于系統(tǒng)cuda版本（最好是等于）。 沒了，就這一個(gè)最重要的原則。舉個(gè)例子進(jìn)行說明：例如經(jīng)過上一步你知道自己系統(tǒng)的cuda版本是11.8，那么你安裝的pytorch對(duì)應(yīng)的cuda版本要小于等于11.8，最好就是是11.8。具體見第4步安裝pytorch。

• 此外，還有pytorch與gcc的兼容性問題，遇到此問題的概率較小，萬一遇到只需要更新一下gcc版本就行，不再贅述。

3.創(chuàng)建基礎(chǔ)的python虛擬環(huán)境

在完成前兩步后，運(yùn)用以下命令創(chuàng)建一個(gè)新的conda環(huán)境，指定名字為farmer，python使用3.9版本。

4.安裝pytorch

在完成第三步的基礎(chǔ)上，我們繼續(xù)。先進(jìn)一步解釋一下第三步創(chuàng)建python虛擬環(huán)境相當(dāng)于做了什么事？此時(shí)其實(shí)就相當(dāng)于將操作系統(tǒng)看做一個(gè)python小區(qū)，小區(qū)有物業(yè)管家比如為python3.7，但你自己在系統(tǒng)里新建了一個(gè)獨(dú)立的小別墅，管家是python3.9，然后在這個(gè)小房間里安裝各種包（添置各種物品，進(jìn)行各種裝修），不會(huì)影響整個(gè)小區(qū)，哪天不爽你可以直接把別墅拆了，同理也可以為不同的項(xiàng)目創(chuàng)建不同的別墅和python管家，安裝不同的包。大家互不影響，也不會(huì)影響到原本的小區(qū)和管家。（大致是這么個(gè)意思）

先在終端激活該虛擬環(huán)境，命令為conda activate farmer(你在上一步中指定的虛擬環(huán)境的名字)，此時(shí)你就可以安裝各種包了，相當(dāng)于給別墅添置各種物品。而所謂的安裝pytorch，其實(shí)也就是安裝幾個(gè)包而已，本質(zhì)上和利用pip安裝個(gè)numpy沒啥區(qū)別。后續(xù)具體步驟如下：

在明確了第一、二步的判斷，并創(chuàng)建、激活虛擬環(huán)境之后，進(jìn)入pytorch官網(wǎng)，如下圖，選擇pytorch版本、操作系統(tǒng)類型、安裝方式（選擇conda）、語(yǔ)言、cuda版本號(hào)，復(fù)制紅框中的命令到vscode的終端（linux、windows終端類似），回車則會(huì)為當(dāng)前python虛擬環(huán)境安裝gpu版pytorch。一般會(huì)有600-1000MB左右，等待安裝完畢，則大功告成。

如果想要選擇特定的pytorch版本： 點(diǎn)擊previous versions of pytorch，以1.12.0為例，看conda下面的就行，wheel下面的不用看。然后同樣的，判斷確認(rèn)自己的操作系統(tǒng)，確認(rèn)cuda版本，復(fù)制相應(yīng)的命令到終端，回車等待下載完畢即可。

這里有一個(gè)問題需要注意：比如前面我們查看了自己電腦的cuda是11.8版本，但在這里并沒有看到相對(duì)應(yīng)的11.8版本，怎么辦呢？這時(shí)候通常只需要選擇相近版本就行，比如11.6或11.7，大概率都是沒有問題的。另外一種就是選擇經(jīng)典版本（cuda10.2和11.3兼容絕大部分pytorch版本），這時(shí)你11.8選擇11.3大概率也是沒有問題的。

終端示例如下，再次強(qiáng)調(diào)，先激活環(huán)境，然后復(fù)制輸入前面說的命令，回車輸入y，等待安裝完畢即可：

完成后可在終端輸入conda list 查看當(dāng)前python虛擬環(huán)境中的所有包，驗(yàn)證pytorch是否安裝成功，如下圖，可以看見安裝的幾個(gè)包對(duì)應(yīng)的python版本和cuda版本。

文章來源地址http://www.zghlxwxcb.cn/news/detail-724483.html

5.總結(jié)提醒

• 在理解本教程的思路后，同理可方便的學(xué)會(huì)其他情況下的安裝思路，并不局限于編輯器是vscode還是pycharm，也不局限于操作系統(tǒng)是linux還是windows。也不局限于大家是在系統(tǒng)默認(rèn)終端，還是vscode或者pycharm里面的終端進(jìn)行操作，無所謂，都可以。
• 最好的步驟是按照上述思路一步步明確進(jìn)行，特別是對(duì)于小白。熟悉后第一、二步均可以快速跳過。一定要先激活對(duì)應(yīng)環(huán)境再進(jìn)行pytorch安裝，否則會(huì)直接裝在系統(tǒng)默認(rèn)的base環(huán)境中，后續(xù)會(huì)引起不必要的麻煩。作者也是踩了很多坑，沒人教一步步才熟悉的，希望大家能少走點(diǎn)彎路！

到了這里，關(guān)于vscode終端安裝pytorch環(huán)境全流程小白版（linux+windows通用版）的文章就介紹完了。如果您還想了解更多內(nèi)容，請(qǐng)?jiān)谟疑辖撬阉鱐OY模板網(wǎng)以前的文章或繼續(xù)瀏覽下面的相關(guān)文章，希望大家以后多多支持TOY模板網(wǎng)！