pytorch環(huán)境搭建詳細(xì)步驟

PyTorch作為一個(gè)廣泛使用的深度學(xué)習(xí)框架，其環(huán)境搭建對(duì)于從事機(jī)器學(xué)習(xí)和深度學(xué)習(xí)研究及開(kāi)發(fā)的人員來(lái)說(shuō)至關(guān)重要。以下將介紹PyTorch環(huán)境搭建的詳細(xì)步驟，包括安裝Anaconda、配置清華鏡像源、創(chuàng)建虛擬環(huán)境、安裝PyTorch及其依賴(lài)庫(kù)、配置PyCharm等。

一、安裝Anaconda

Anaconda是一個(gè)開(kāi)源的Python和R語(yǔ)言的分布式版本控制系統(tǒng)，旨在簡(jiǎn)化包管理和部署。它包含了conda、Python等180多個(gè)科學(xué)包及其依賴(lài)項(xiàng)，非常適合用于科學(xué)計(jì)算（數(shù)據(jù)科學(xué)、機(jī)器學(xué)習(xí)應(yīng)用、大數(shù)據(jù)處理和可視化）。

1. 下載Anaconda
   • 前往Anaconda官網(wǎng)下載Anaconda個(gè)人版。由于官網(wǎng)下載速度可能較慢，推薦使用清華鏡像源進(jìn)行下載，以提高下載速度。
   • 選擇適合您操作系統(tǒng)的版本進(jìn)行下載，例如Windows x86_64版本。
2. 安裝Anaconda
   • 下載完成后，雙擊安裝程序進(jìn)行安裝。在安裝過(guò)程中，請(qǐng)遵循安裝向?qū)У闹甘具M(jìn)行操作。
   • 注意在安裝過(guò)程中可以選擇是否將Anaconda添加到系統(tǒng)PATH中，建議選擇添加，以便在命令行中直接調(diào)用conda命令。
3. 驗(yàn)證安裝
   • 安裝完成后，打開(kāi)命令行或Anaconda Prompt，輸入conda --version來(lái)驗(yàn)證Anaconda是否成功安裝。

二、配置清華鏡像源

由于PyTorch的服務(wù)器在國(guó)外，直接下載可能速度較慢，因此推薦配置清華鏡像源以加速下載過(guò)程。

1. 打開(kāi)Anaconda Prompt
   • 在Windows系統(tǒng)中，可以通過(guò)開(kāi)始菜單找到并打開(kāi)Anaconda Prompt。
2. 添加清華鏡像源
   • 在Anaconda Prompt中輸入以下命令來(lái)添加清華鏡像源：

conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/free/  
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main/  
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/pytorch/  
# 對(duì)于舊版本的win-64系統(tǒng)，可能需要添加以下鏡像源  
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/peterjc123/  
conda config --set show_channel_urls yes

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

使用conda創(chuàng)建虛擬環(huán)境可以避免不同項(xiàng)目之間的包版本沖突。

1. 查看已存在的虛擬環(huán)境
   • 在Anaconda Prompt中輸入conda info --envs或conda env list來(lái)查看當(dāng)前已存在的虛擬環(huán)境列表。
2. 創(chuàng)建新的虛擬環(huán)境
   • 假設(shè)我們要?jiǎng)?chuàng)建一個(gè)名為pytorch的虛擬環(huán)境，并指定Python版本為3.9（具體版本可根據(jù)需要選擇），可以使用以下命令：

conda create -n pytorch python=3.9

• 等待命令執(zhí)行完成，虛擬環(huán)境即創(chuàng)建成功。

3. 激活虛擬環(huán)境
   • 使用conda activate pytorch命令來(lái)激活剛剛創(chuàng)建的虛擬環(huán)境。激活后，命令行提示符前將顯示虛擬環(huán)境名稱(chēng)，表示當(dāng)前處于該虛擬環(huán)境中。
4. 退出虛擬環(huán)境
   • 使用conda deactivate命令可以退出當(dāng)前激活的虛擬環(huán)境。

四、安裝PyTorch及其依賴(lài)庫(kù)

在虛擬環(huán)境中安裝PyTorch及其依賴(lài)庫(kù)（如torchvision和cudatoolkit）可以確保這些庫(kù)僅在該虛擬環(huán)境中可用，避免影響其他項(xiàng)目。

1. 安裝PyTorch
   • 根據(jù)您的需求（是否需要GPU支持、CUDA版本等），選擇合適的PyTorch安裝命令。以下是一個(gè)示例命令，用于安裝支持CUDA 10.2的PyTorch版本（請(qǐng)根據(jù)實(shí)際情況調(diào)整）：

conda install pytorch torchvision cudatoolkit=10.2 -c pytorch

• 注意：由于已經(jīng)配置了清華鏡像源，因此這里的-c pytorch可以省略，conda會(huì)自動(dòng)從清華鏡像源中查找并下載PyTorch及其依賴(lài)庫(kù)。

2. 驗(yàn)證安裝
   • 安裝完成后，可以在Python解釋器中通過(guò)導(dǎo)入torch和torchvision來(lái)驗(yàn)證PyTorch是否成功安裝。例如，在Anaconda Prompt中激活虛擬環(huán)境后，輸入python進(jìn)入Python解釋器，然后輸入以下命令：

import torch  
import torchvision  
print(torch.__version__)  
print(torchvision. **version** )
# 檢查CUDA是否可用（如果安裝了GPU版本的PyTorch）  
 if torch.cuda.is_available():  
     print("CUDA is available. GPU accelerated PyTorch.")  
 else:  
     print("CUDA is not available. Running on CPU.")

這些命令將輸出PyTorch和torchvision的版本號(hào)，并檢查CUDA是否可用。如果CUDA可用，則表明GPU版本的PyTorch已正確安裝。

五、配置PyCharm（或其他IDE）

雖然使用命令行進(jìn)行開(kāi)發(fā)是完全可行的，但許多開(kāi)發(fā)者更喜歡使用集成開(kāi)發(fā)環(huán)境（IDE）如PyCharm來(lái)提高開(kāi)發(fā)效率。以下是在PyCharm中配置PyTorch環(huán)境的步驟：

1. 打開(kāi)PyCharm
   • 啟動(dòng)PyCharm，并選擇“Create New Project”來(lái)創(chuàng)建一個(gè)新項(xiàng)目。
2. 配置項(xiàng)目解釋器
   • 在項(xiàng)目創(chuàng)建向?qū)е?，找到“Project Interpreter”部分。
   • 選擇“Existing environment”選項(xiàng)，然后點(diǎn)擊右側(cè)的“...”按鈕來(lái)瀏覽并選擇您之前創(chuàng)建的PyTorch虛擬環(huán)境。PyCharm將自動(dòng)檢測(cè)該環(huán)境中的Python解釋器和已安裝的包。
3. 創(chuàng)建虛擬環(huán)境（可選）
   • 如果您還沒(méi)有在PyCharm外部創(chuàng)建虛擬環(huán)境，也可以在PyCharm中直接創(chuàng)建。在“Project Interpreter”部分，選擇“New environment using”選項(xiàng)，然后選擇“Conda”作為環(huán)境類(lèi)型，并輸入新的環(huán)境名稱(chēng)。之后，PyCharm將引導(dǎo)您完成虛擬環(huán)境的創(chuàng)建和PyTorch的安裝。
4. 完成項(xiàng)目創(chuàng)建
   • 配置好項(xiàng)目解釋器后，點(diǎn)擊“Create”按鈕完成項(xiàng)目創(chuàng)建?，F(xiàn)在，您的PyCharm項(xiàng)目將使用指定的PyTorch虛擬環(huán)境進(jìn)行開(kāi)發(fā)。

六、進(jìn)一步配置（可選）

根據(jù)您的具體需求，您可能還需要進(jìn)行一些額外的配置，例如：

• 安裝其他Python包 ：通過(guò)conda install或pip install命令在虛擬環(huán)境中安裝其他必要的Python包。
• 配置Jupyter Notebook ：如果您喜歡使用Jupyter Notebook進(jìn)行交互式編程，可以在虛擬環(huán)境中安裝Jupyter Notebook，并通過(guò)PyCharm的Jupyter Notebook集成插件或直接在命令行中啟動(dòng)Jupyter Notebook來(lái)編寫(xiě)和運(yùn)行代碼。
• 設(shè)置環(huán)境變量 ：在某些情況下，您可能需要設(shè)置環(huán)境變量來(lái)確保程序能夠正確訪(fǎng)問(wèn)外部資源或庫(kù)。這可以通過(guò)在Anaconda Prompt中設(shè)置環(huán)境變量，或在PyCharm的項(xiàng)目設(shè)置中配置環(huán)境變量來(lái)實(shí)現(xiàn)。

七、總結(jié)

通過(guò)以上步驟，您應(yīng)該能夠成功搭建一個(gè)包含PyTorch的Python開(kāi)發(fā)環(huán)境。這個(gè)環(huán)境將使用Anaconda進(jìn)行包管理和虛擬環(huán)境管理，通過(guò)配置清華鏡像源來(lái)加速下載過(guò)程，并通過(guò)PyCharm等IDE提供高效的開(kāi)發(fā)體驗(yàn)。