它不僅功能強大,而且易于上手,尤其適合研究和開發(fā)
然而,對于許多初學者來說,在Linux系統(tǒng)下安裝PyTorch可能是一項挑戰(zhàn)
本文將詳細介紹如何在Linux系統(tǒng)下高效、穩(wěn)定地安裝PyTorch,幫助大家快速搭建深度學習環(huán)境
一、準備工作 在開始安裝之前,確保你的系統(tǒng)滿足以下基本要求: - 操作系統(tǒng):Linux(推薦使用Ubuntu或其他主流發(fā)行版)
- Python版本:建議使用Python 3.7或更高版本
Python是PyTorch的基石,合適的版本能夠確保兼容性和性能
- 硬件支持:根據(jù)你的需求,確定是否需要GPU支持
如果計算機配備了NVIDIA GPU,并且希望加速深度學習計算,可以安裝GPU版本的PyTorch
此外,建議使用虛擬環(huán)境來管理項目依賴,以避免包沖突
Anaconda是一個流行的虛擬環(huán)境管理工具,它不僅能方便地創(chuàng)建和管理虛擬環(huán)境,還能提供大量的科學計算包
二、安裝Anaconda 1.下載Anaconda 訪問Anaconda官網(wǎng),根據(jù)你的系統(tǒng)選擇對應的版本
Anaconda會自動檢測你的操作系統(tǒng),并推薦合適的版本
你也可以選擇從國內(nèi)鏡像站(如清華大學開源軟件鏡像站)下載,以提高下載速度
使用以下命令在Linux終端下載Anaconda: bash wget https://mirrors.tuna.tsinghua.edu.cn/anaconda/archive/Anaconda3-2023.03-Linux-x86_64.sh 2.安裝Anaconda 下載完成后,執(zhí)行以下命令進行安裝: bash bash Anaconda3-2023.03-Linux-x86_64.sh 閱讀并接受安裝協(xié)議,確認安裝位置(默認即可),等待安裝完成
3.配置環(huán)境變量 安裝完成后,需要配置環(huán)境變量以便系統(tǒng)能夠正確找到Anaconda
編輯`~/.bashrc`文件,添加以下內(nèi)容: bash export PATH=/root/anaconda3/bin:$PATH 這里的路徑根據(jù)實際情況調(diào)整 source ~/.bashrc 4.初始化Anaconda 執(zhí)行以下命令,使conda的基礎環(huán)境在啟動時不被激活: bash conda init conda config --setauto_activate_base false 5.配置conda源 為了提高包下載和更新的速度,建議使用國內(nèi)鏡像源
例如,使用清華鏡像源: bash conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/conda-forge/ 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/ conda config --setshow_channel_urls yes 三、創(chuàng)建虛擬環(huán)境 使用Anaconda創(chuàng)建虛擬環(huán)境可以隔離不同項目的依賴項,避免項目之間的沖突
1.創(chuàng)建虛擬環(huán)境 執(zhí)行以下命令創(chuàng)建一個新的虛擬環(huán)境,例如名為`pytorch`的環(huán)境,并指定Python版本為3.8: bash co