pytorch安装 linux
时间: 2023-04-16 18:02:42 浏览: 73
在Linux上安装PyTorch,可以通过以下步骤完成:
1. 安装Anaconda或Miniconda,建议使用Anaconda,因为它包含了更多的Python包和工具。
2. 创建一个新的conda环境,可以使用以下命令:
```
conda create --name pytorch_env python=3.7
```
这将创建一个名为pytorch_env的新环境,并使用Python 3.7版本。
3. 激活新的conda环境,可以使用以下命令:
```
conda activate pytorch_env
```
4. 安装PyTorch,可以使用以下命令:
```
conda install pytorch torchvision torchaudio cpuonly -c pytorch
```
这将安装PyTorch、TorchVision和TorchAudio,并且只使用CPU版本。如果你想使用GPU版本,可以将cpuonly替换为cudatoolkit,例如:
```
conda install pytorch torchvision torchaudio cudatoolkit -c pytorch
```
5. 安装完成后,可以在Python中导入PyTorch并开始使用它:
```
import torch
```
如果没有出现任何错误,说明PyTorch已经成功安装。
希望这些步骤能够帮助你在Linux上安装PyTorch。