在Linux上運行pytorch代碼,您需要先安裝Python和PyTorch
-
安裝Python: 打開終端,輸入以下命令來安裝Python 3(如果尚未安裝):
sudo apt update sudo apt install python3
-
安裝pip: pip是Python的包管理器,用于安裝和管理Python軟件包。在終端中輸入以下命令來安裝pip:
sudo apt install python3-pip
-
創建虛擬環境(可選): 虛擬環境可以讓您在不同的項目中使用不同版本的Python和庫。要創建一個虛擬環境,請按照以下步驟操作:
a. 安裝virtualenv:
pip3 install virtualenv
b. 創建一個新的虛擬環境:
virtualenv my_pytorch_env
c. 激活虛擬環境:
source my_pytorch_env/bin/activate
d. 現在,您的虛擬環境已激活。您可以在此環境中安裝PyTorch和其他所需的庫。
-
安裝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
-
運行PyTorch代碼: 在終端中,導航到包含您的PyTorch代碼的目錄,然后使用Python運行代碼。例如:
cd /path/to/your/pytorch_project python3 your_script.py
如果您使用了虛擬環境,請確保在運行代碼之前激活它。
現在,您已經成功在Linux上安裝了PyTorch并可以運行PyTorch代碼了。