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