Linux PyTorch环境如何搭建 - 问答

在Linux系统上搭建PyTorch环境，你可以选择使用Anaconda或者直接使用pip安装。以下是两种常见的方法：

Anaconda是一个流行的Python数据科学平台，它提供了一个方便的方式来管理包和环境。

下载并安装Anaconda：
- 访问Anaconda官网下载适合你系统的安装包。
- 按照安装向导完成安装。
创建新的conda环境（可选）：
```
conda create -n pytorch_env python=3.8
```
这里pytorch_env是你新环境的名字，python=3.8是你想要使用的Python版本。
激活新环境：
```
conda activate pytorch_env
```
安装PyTorch：
- 访问PyTorch官网，选择适合你系统的安装命令。
- 例如，如果你使用的是CUDA 11.3，可以运行：
```
conda install pytorch torchvision torchaudio cudatoolkit=11.3 -c pytorch
```
- 如果你不使用GPU，可以运行：
```
conda install pytorch torchvision torchaudio cpuonly -c pytorch
```

如果你不想使用Anaconda，可以直接使用pip来安装PyTorch。

确保pip是最新的：
```
pip install --upgrade pip
```
安装PyTorch：
- 同样访问PyTorch官网，选择适合你系统的安装命令。
- 例如，如果你使用的是CUDA 11.3，可以运行：
```
pip install torch torchvision torchaudio
```
- 如果你不使用GPU，可以运行：
```
pip install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cpu
```

无论使用哪种方法安装，都可以通过以下命令来验证PyTorch是否安装成功：

import torch
print(torch.__version__)
print(torch.cuda.is_available())  # 如果使用GPU，应该返回True

如果以上代码没有报错，并且显示了PyTorch的版本号，以及GPU是否可用的信息，那么恭喜你，PyTorch已经成功安装！

0 赞

0 踩