linux安装PYTORCH
时间: 2025-01-04 15:31:55 浏览: 8
### 如何在 Linux 操作系统上安装 PyTorch
#### 准备工作
为了确保顺利安装 PyTorch,在开始之前需确认已正确设置好 Linux 系统环境。对于 Ubuntu 20.04 来说,创建合适的分区非常重要,特别是 efi 系统分区不可或缺[^2]。
#### 安装 Anaconda 或 Miniconda
推荐先安装 Anaconda 或者更轻量级的选择 Miniconda。Anaconda 提供了一个完整的 Python 数据科学平台,而 Miniconda 则只包含了 conda 包管理和 Python 解释器本身。这两种方式都可以简化后续依赖项管理并减少冲突的可能性。
```bash
# 下载Miniconda脚本 (假设使用Python3版本)
wget https://repo.anaconda.com/miniconda/Miniconda3-latest-Linux-x86_64.sh
# 运行安装程序
bash Miniconda3-latest-Linux-x86_64.sh
```
#### 创建虚拟环境
建议为不同的项目分别建立独立的虚拟环境来隔离各个项目的包依赖关系:
```bash
# 使用conda命令创建名为myenv的新环境,并指定python版本
conda create --name myenv python=3.9
# 激活新创建的环境
conda activate myenv
```
#### 安装 PyTorch
访问官方提供的快速启动页面获取适合当前系统的具体安装指令。通常情况下可以通过 pip 或者 conda 方式来进行安装。这里给出基于 CUDA 支持 GPU 加速的例子;如果不具备 NVIDIA 显卡,则可以选择 CPU 版本[^4]。
##### 使用 Conda 安装(含CUDA支持)
```bash
conda install pytorch torchvision torchaudio cudatoolkit=11.3 -c pytorch
```
##### 使用 Pip 安装(仅CPU版)
```bash
pip install torch torchvision torchaudio
```
完成上述操作之后就可以验证安装是否成功了。可以尝试导入 `torch` 库并在交互模式下运行简单的张量运算测试。
```python
import torch
print(torch.__version__)
x = torch.rand(5, 3)
print(x)
```
如果一切正常的话应该会看到打印出来的 PyTorch 版本号以及随机生成的一个矩阵数据。
阅读全文