前言

因為我之前安裝環(huán)境花費了不少時間，也是網(wǎng)上搜了很多，想著自己寫一個幫助大家快速安裝，少走彎路，快速開啟深度之路。

一、判斷是否有Nvidia(英偉達(dá)顯卡)

下面我以我的臺式電腦windows10系統(tǒng)為例

第一步：右鍵開始菜單，在彈出選項中選擇任務(wù)管理器

進(jìn)來就是這個頁面，這臺是沒有GPU的

這臺是有GPU的

首先點性能選項，然后點GPU。在右上方會顯示GPU名稱，只有帶NVIDIA的英偉達(dá)顯卡的電腦才能安裝GPU版本，否則其他的就只能安裝CPU版本。我會寫兩部分，各位根據(jù)自己的電腦選擇觀看就好

二、CPU版

2.1 安裝Anaconda

首先去官網(wǎng)下載Anaconda。

點擊download，下載的就是最新版本的。

下載完成后，直接運行下步就行

注意到路徑這塊，最好是全英文。

接下來根據(jù)個人喜好勾選，可以不用管直接下一步安裝好就行。

那怎么驗證是否安裝好了呢？

在開始菜單下，找到帶有綠色圖標(biāo)的Anaconda,點擊就可以打開Anaconda的圖形化界面

點擊環(huán)境，看到base（root），這個就是默認(rèn)的一個虛擬環(huán)境，代表我們安裝成功了。

2.2 創(chuàng)建虛擬環(huán)境

為什么需要學(xué)會這個呢？其實剛剛安裝好Anaconda會自帶一個基礎(chǔ)環(huán)境。但是我們后續(xù)的項目每一個需要的安裝包不同，為了避免沖突，所以我們可以為每一個項目配置一個虛擬環(huán)境，這樣就不相互打擾了。

上面兩個是Anaconda命令窗口的位置，隨便點擊那個都可以。

conda env list

輸入上面的命令就可以查看當(dāng)前有多少虛擬環(huán)境。

• 利用 conda create 指令創(chuàng)建新的虛擬環(huán)境

? conda create –n 虛擬環(huán)境名字 python=版本

演示如下：

會出現(xiàn)如下圖

輸入y就行，然后就是等待安裝。好了以后用指令檢查一下，如下，就是成功的。

2.3安裝pytorch

等等，安裝前這里還有一個問題，因為我們的安裝指令都是從國外的通道下載，下載速度會非常慢。

所以我們想要下載快還是得用國內(nèi)的鏡像地址。

conda config --show

輸入上面的指令可以查看我們的通道地址

現(xiàn)在我們的環(huán)境中通道是defaults，也就是說它會從國外給我下載包。

也可以通過下面的指令看到

conda config --get

下面是一些鏡像地址

關(guān)于加通道地址這塊，推薦是下載時可以指定安裝通道，而不是直接去改配置文件，這樣出現(xiàn)bug的幾率比較小。下面我也會提供這種方法不過不推薦。

• 往配置文件中加通道地址

conda config --add channels 通道地址

那么如何驗證是否加上呢？

還是用上面的get指令

這里也會標(biāo)注，優(yōu)先使用我們新加上的通道，如果新通道沒有，再使用默認(rèn)通道去找。

• 萬一你想刪除剛剛的通道怎么辦呢？用下面這個指令

conda config --remove channels 通道地址

• 安裝pytorch前想清楚一個問題，想把pytorch安裝到那個環(huán)境里。

默認(rèn)是在基礎(chǔ)環(huán)境下（base），比如我現(xiàn)在想安裝到y(tǒng)olov8這個虛擬環(huán)鏡

conda activate yolov8

• 輸入上面命令進(jìn)入想安裝pytorch的虛擬環(huán)境，上面就是想在yolov8這個虛擬環(huán)境中安裝pytorch

進(jìn)入pytorch官網(wǎng)

conda install pytorch torchvision torchaudio cpuonly -c pytorch//這種就是直接從國外下載了

下面這種是鏡像加速版

conda install pytorch torchvision torchaudio cpuonly –c 鏡像地址

以清華鏡像源為例：

出現(xiàn)下圖，輸入y即可

出現(xiàn)下圖就是安裝好了

2.4 驗證pytorch是否安裝成功

激活對應(yīng)的虛擬環(huán)境（你安裝Pytorch的虛擬環(huán)境）

conda activate 虛擬環(huán)境名

輸入conda list，看有沒有pytorch或者torch

比如我這里就是pytorch

輸入python

輸入 import torch

輸入 torch.cuda.is_available()，這個命令是檢查我們電腦是否可以用GPU，因為沒有，當(dāng)然就是false。

如果顯示False，就說明我們這個PyTorch安裝成功了

三、GPU版

3.1 安裝Anaconda

首先去官網(wǎng)下載Anaconda。

點擊download，下載的就是最新版本的。

下載完成后，直接運行下步就行

注意到路徑這塊，最好是全英文。

接下來根據(jù)個人喜好勾選，可以不用管直接下一步安裝好就行。

那怎么驗證是否安裝好了呢？

在開始菜單下，找到帶有綠色圖標(biāo)的Anaconda,點擊就可以打開Anaconda的圖形化界面

點擊環(huán)境，看到base（root），這個就是默認(rèn)的一個虛擬環(huán)境，代表我們安裝成功了。

3.2 創(chuàng)建虛擬環(huán)境

為什么需要學(xué)會這個呢？其實剛剛安裝好Anaconda會自帶一個基礎(chǔ)環(huán)境。但是我們后續(xù)的項目每一個需要的安裝包不同，為了避免沖突，所以我們可以為每一個項目配置一個虛擬環(huán)境，這樣就不相互打擾了。

上面兩個是Anaconda命令窗口的位置，隨便點擊那個都可以。

conda env list

輸入上面的命令就可以查看當(dāng)前有多少虛擬環(huán)境。

• 利用 conda create 指令創(chuàng)建新的虛擬環(huán)境

? conda create –n 虛擬環(huán)境名字 python=版本

演示如下：

會出現(xiàn)如下圖

輸入y就行，然后就是等待安裝。好了以后用指令檢查一下，如下，就是成功的。

3.3 CUDA的準(zhǔn)備工作

要想使用GPU得先了解自己電腦的GPU。

鼠標(biāo)右鍵開始菜單，點擊任務(wù)管理器，進(jìn)入性能中查看自己電腦GPU型號。

首先安裝最新的顯卡驅(qū)動,進(jìn)入官網(wǎng)選擇適合自己電腦的配置，我的配置如下。

下載完直接安裝就行。

然后是查看自己電腦驅(qū)動的版本，win+r，輸入cmd，再輸入下面的命令

nvidia-smi

我的CUDA Driver版本就是11.6，表示的是驅(qū)動所能支持的最大運行API版本就是11.6。我如果要安裝CUDA Runtime Version（運行版本），要保證CUDA Driver 版本 >= CUDA Runtime 版本，也就是11.6及以前的。

其實裝了Anaconda之后Anaconda會提供一個cudatoolkit工具包，同樣包含了CUDA的運行API，可以用來替代官方CUDA的CUDA Toolkit。查看官方CUDA的運行API版本。

nvcc -V

可以看到是11.1，也就是說我的電腦安裝pytorch時最好選擇11.1-11.6版本的cuda。

3.4 安裝pytorch

等等，安裝前這里還有一個問題，因為我們的安裝指令都是從國外的通道下載，下載速度會非常慢。

所以我們想要下載快還是得用國內(nèi)的鏡像地址。

conda config --show

輸入上面的指令可以查看我們的通道地址

現(xiàn)在我們的環(huán)境中通道是defaults，也就是說它會從國外給我下載包。

也可以通過下面的指令看到

conda config --get

下面是一些鏡像地址

添加通道

conda config --add channels 通道地址

那么如何驗證是否加上呢？

還是用上面的get指令

這里也會標(biāo)注，優(yōu)先使用我們新加上的通道，如果新通道沒有，再使用默認(rèn)通道去找。

萬一你想刪除剛剛的通道怎么辦呢？如下

conda config --remove channels 通道地址

關(guān)于加通道地址這塊，推薦是下載時可以指定安裝通道，而不是直接去改配置文件，這樣出現(xiàn)bug的幾率比較小。不推薦直接去改配置文件這種方法。

安裝pytorch前想清楚一個問題，想把pytorch安裝到那個環(huán)境里。

默認(rèn)是在基礎(chǔ)環(huán)境下（base），比如我現(xiàn)在想安裝到y(tǒng)olov8這個虛擬環(huán)鏡

conda activate yolov8

輸入上面命令進(jìn)入想安裝pytorch的虛擬環(huán)境，上面就是想在yolov8這個虛擬環(huán)境中安裝pytorch

進(jìn)入pytorch官網(wǎng)

3.4.1 conda安裝法（不推薦）

進(jìn)來發(fā)現(xiàn)cuda不符，就去以前的版本找。

大家根據(jù)自己的環(huán)境需求找，我最后選擇這個。如果網(wǎng)絡(luò)還可以，就用官網(wǎng)的下載就行。

conda install pytorch==1.12.1 torchvision==0.13.1 torchaudio==0.12.1 cudatoolkit=11.3 -c pytorch

下面介紹一種鏡像的下載。

第一步：因為cudatoolkit下載很慢，所以也單獨指定一個鏡像

Conda install cudatoolkit=版本 –c 鏡像地址

第二步：

conda install pytorch torchvision torchaudio  -c 鏡像地址

這樣后續(xù)就好了。

3.4.2 pip安裝法（推薦）

進(jìn)入pytorch官網(wǎng)

如果你的電腦可以裝11.7或者11.8就直接粘貼復(fù)制到anaconda窗口，不行就像我一樣去以前的版本找。

3.5 驗證pytorch是否安裝成功

激活對應(yīng)的虛擬環(huán)境（你安裝Pytorch的虛擬環(huán)境）

conda activate 虛擬環(huán)境名

輸入pip list或者conda list，看有沒有pytorch或者torch

比如我這里就是torch

3. 輸入python

4. 輸入 import torch

6. 輸入 torch.cuda.is_available()，這個命令是檢查我們pytorch的GPU能否用。

7. 如果顯示True，就說明我們這個PyTorch安裝成功了

四、 pycharm安裝與配置

首先進(jìn)入pycharm官網(wǎng)

點擊download

下載社區(qū)版就夠用了。

注意路徑全英文。

我只勾了.py,其他的根據(jù)自己的需要來勾選。直接下一步等待安裝完成。

裝完后在開始菜單下，找到pycharm點開。

我喜歡亮一點，在這里調(diào)背景色。

開始新建項目

自己指定項目的位置。

因為我們前面已經(jīng)配好了環(huán)境，直接用我們配好的，點擊add去找我們配置的位置

從左往右按順序點擊，找到自己anaconda安裝的位置。

用我們已經(jīng)創(chuàng)建好的環(huán)境，比如我這里是yolov8

點擊create，項目就創(chuàng)建好了。

創(chuàng)建一個文件測試一下

import torch
print(torch.cuda.is_available())

鼠標(biāo)右健后，點擊run就行

這就代表成功了，注意CPU版本這里應(yīng)該是False。

總結(jié)

到此這篇關(guān)于2023最新pytorch快速安裝指南的文章就介紹到這了,更多相關(guān)最新pytorch安裝內(nèi)容請搜索腳本之家以前的文章或繼續(xù)瀏覽下面的相關(guān)文章希望大家以后多多支持腳本之家！

最新評論