溫馨提示×

如何在Linux上運行PyTorch代碼

小樊
48
2025-06-05 04:57:49
欄目: 智能運維

在Linux上運行PyTorch代碼,您需要先安裝Python和PyTorch

  1. 安裝Python: 打開終端,輸入以下命令來安裝Python 3(如果尚未安裝):

    sudo apt update
    sudo apt install python3
    
  2. 安裝pip: pip是Python的包管理器,用于安裝和管理Python軟件包。在終端中輸入以下命令來安裝pip:

    sudo apt install python3-pip
    
  3. 創建虛擬環境(可選): 虛擬環境可以讓您在不同的項目中使用不同版本的Python和庫。要創建一個虛擬環境,請按照以下步驟操作:

    a. 安裝virtualenv

    pip3 install virtualenv
    

    b. 創建一個新的虛擬環境:

    virtualenv my_pytorch_env
    

    c. 激活虛擬環境:

    source my_pytorch_env/bin/activate
    

    d. 現在,您的虛擬環境已激活。您可以在此環境中安裝PyTorch和其他所需的庫。

  4. 安裝PyTorch: 根據您的系統和CUDA版本,選擇合適的PyTorch安裝命令。訪問PyTorch官方網站(https://pytorch.org/get-started/locally/)以獲取適用于您系統的安裝命令。例如,對于沒有CUDA支持的CPU版本,您可以使用以下命令:

    pip3 install torch torchvision torchaudio
    

    對于具有CUDA支持的GPU版本,請根據您的CUDA版本選擇相應的命令。例如,對于CUDA 11.3,您可以使用以下命令:

    pip3 install torch torchvision torchaudio --extra-index-url https://download.pytorch.org/whl/cu113
    
  5. 運行PyTorch代碼: 在終端中,導航到包含您的PyTorch代碼的目錄,然后使用Python運行代碼。例如:

    cd /path/to/your/pytorch_project
    python3 your_script.py
    

    如果您使用了虛擬環境,請確保在運行代碼之前激活它。

現在,您已經成功在Linux上安裝了PyTorch并可以運行PyTorch代碼了。

0
亚洲午夜精品一区二区_中文无码日韩欧免_久久香蕉精品视频_欧美主播一区二区三区美女