Linux服务器下安装pytorch

### 如何在Linux服务器上安装PyTorch #### 使用pip安装特定版本的PyTorch 如果需要安装指定路径中的 `.whl` 文件，可以通过 `pip` 工具完成此操作。例如，在 `/home/lihao/torch` 目录下存在名为 `torch-1.7.1-cp39-cp39-linux_aarch64.whl` 的文件，则可运行以下命令来安装该包[^1]： ```bash pip install /home/lihao/torch/torch-1.7.1-cp39-cp39-linux_aarch64.whl ``` #### 配置Anaconda环境并安装PyTorch 对于更复杂的项目需求，推荐使用 Anaconda 来管理 Python 环境以及依赖项。以下是具体步骤： 1. **下载并安装Anaconda** 下载适用于 Linux 平台的 Anaconda 安装脚本，并执行安装程序： ```bash bash Anaconda3-5.0.1-Linux-x86_64.sh ``` 在安装过程中需确认是否将安装路径添加至环境变量中。 2. **使更改生效** 执行以下命令让新设置立即可用： ```bash source ~/.bashrc ``` 3. **创建虚拟环境** 创建一个新的 Conda 虚拟环境（假设名称为 `myenv`），并指定所需的 Python 版本： ```bash conda create -n myenv python=3.9 ``` 4. **激活虚拟环境** 启动刚刚创建好的虚拟环境以便后续操作在此环境中进行： ```bash source activate myenv ``` 5. **通过Conda安装PyTorch及相关组件** 如果目标是 GPU 支持版 PyTorch (基于 CUDA Toolkit 10.2)，则应采用如下方式安装[^5]: ```bash conda install pytorch==1.8.0 torchvision==0.9.0 torchaudio==0.8.0 cudatoolkit=10.2 -c pytorch ``` #### 利用官方源直接安装最新稳定版PyTorch 当不需要特别定制化安装时，也可以借助 PyTorch 提供的标准方法快速部署其环境。下面展示的是针对 CPU 和 GPU 不同情况下的两种典型做法之一——仅限于 CPU 的场景[^4]: ```bash pip install torch===1.5.1 torchvision===0.6.1 -f https://download.pytorch.org/whl/torch_stable.html -i https://pypi.douban.com/simple ``` 完成后即可依据个人研究方向开展进一步工作[^2]。