PyTorch Linux环境如何配置

在Linux系统上配置PyTorch环境通常涉及以下几个步骤：

1. 安装Anaconda（推荐）

• 下载并安装Anaconda： 访问Anaconda官网下载适合你Linux系统的安装包，并按照安装向导完成安装。

• 创建并激活新的conda环境（可选但推荐）：

  conda create -n pytorch_env python=3.8
  conda activate pytorch_env
  

  这将创建一个名为pytorch_env的新环境，并激活它。你可以根据需要更改环境名称和Python版本。

2. 安装PyTorch

• 使用conda安装PyTorch（推荐）： 访问PyTorch官网获取适合你系统的安装命令。例如，对于带有CUDA支持的版本，可以使用以下命令：

  conda install pytorch torchvision torchaudio cudatoolkit=11.3 -c pytorch
  

  如果你没有NVIDIA GPU或不想使用CUDA，可以安装CPU版本：

  conda install pytorch torchvision torchaudio cpuonly -c pytorch
  

• 使用pip安装PyTorch（如果使用Anaconda则这一步可选）： 如果你选择不使用Anaconda，可以直接使用pip安装PyTorch。首先确保你已经安装了Python和pip：

  sudo apt update
  sudo apt install python3 python3-pip
  

  然后安装PyTorch：

  pip3 install torch torchvision torchaudio
  

  或者，如果你需要CUDA支持：

  pip3 install torch torchvision torchaudio --extra-index-url https://download.pytorch.org/whl/cu113
  

3. 验证安装

• 检查PyTorch版本： 在激活的conda环境中，运行以下Python代码来验证PyTorch是否安装成功：

  import torch
  print(torch.__version__)
  print(torch.cuda.is_available())  # 应该返回True如果你安装的是GPU版本
  

4. 配置虚拟环境（可选）

• 创建和激活虚拟环境： 为了更好地管理依赖包，你可以使用Python的venv模块创建虚拟环境：

  python3 -m venv pytorch_env
  source pytorch_env/bin/activate
  

  在虚拟环境中安装PyTorch，并验证安装。

5. 配置开发环境（可选）

• 安装开发工具：
  • PyCharm：面向专业开发者的Python IDE，可以下载社区版和专业版。
  • 其他编辑器：如VS Code等。

6. 配置PyTorch在IDE中（可选）

• 在PyCharm中配置PyTorch虚拟环境：
  • 打开PyCharm，创建一个新项目。
  • 进入File -> Settings -> Project: pytorch_learning - Python Interpreter。
  • 点击Add Interpreter，选择你创建的pytorch_env虚拟环境中的Python解释器。