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版本，确保其与您的硬件和软件环境兼容。