Ubuntu下PyTorch版本選擇

在Ubuntu下選擇PyTorch版本時，需要考慮多個因素，包括CUDA版本、Python版本、系統穩定性以及社區支持等。以下是一些推薦的版本和安裝步驟：

推薦版本

1. CUDA 11.3：

   • PyTorch 1.10：適用于CUDA 11.3，推薦與Ubuntu 20.04或22.04搭配使用。
   • PyTorch 1.12.1：適用于CUDA 11.4，可以在Ubuntu 22.04上使用，選擇此版本時安裝命令為：

     conda install pytorch=1.12.1 torchvision=0.13.1 torchaudio=0.12.1 cudatoolkit=11.3 -c pytorch
     

2. CUDA 12.1：

   • PyTorch 2.4.0：適用于CUDA 12.1，推薦與Ubuntu 22.04搭配使用。

安裝步驟

使用Conda安裝PyTorch

1. 創建并激活Conda環境：

   conda create -n myenv python=3.8
   conda activate myenv
   

2. 安裝PyTorch：

   • 支持CUDA 11.3：

     conda install pytorch=1.12.1 torchvision=0.13.1 torchaudio=0.12.1 cudatoolkit=11.3 -c pytorch
     

   • 支持CUDA 12.1：

     conda install pytorch=2.4.0 torchvision=0.19.0 torchaudio=2.4.0 cudatoolkit=12.1 -c pytorch -c nvidia
     

驗證安裝

安裝完成后，可以通過以下命令驗證PyTorch是否安裝成功：

python -c "import torch; print(torch.__version__)"
python -c "import torch; print(torch.cuda.is_available())"

通過這些步驟和建議，您可以在Ubuntu下選擇合適的PyTorch版本，確保其與您的硬件和軟件環境兼容。