linux安装pytorch
时间: 2023-10-13 11:19:56 浏览: 40
要在Linux上安装PyTorch,请按照以下步骤进行操作:
1. 安装Anaconda或Miniconda:在https://www.anaconda.com/products/individual或https://docs.conda.io/en/latest/miniconda.html上下载适合您的操作系统的版本,并安装。
2. 创建一个新的conda环境:打开终端并输入以下命令:
```
conda create --name pytorch_env python=3.7
```
这将创建一个名为pytorch_env的新conda环境,并使用Python 3.7。
3. 激活conda环境:运行以下命令以激活创建的conda环境:
```
conda activate pytorch_env
```
4. 安装PyTorch:运行以下命令以在conda环境中安装PyTorch:
```
conda install pytorch torchvision torchaudio cpuonly -c pytorch
```
这将安装适用于CPU的PyTorch版本。如果您使用的是GPU,则可以使用以下命令安装适用于GPU的PyTorch版本:
```
conda install pytorch torchvision torchaudio cudatoolkit=10.2 -c pytorch
```
5. 验证安装:在激活的conda环境中运行Python解释器,并导入PyTorch:
```
python
import torch
```
如果没有错误消息,则说明PyTorch已成功安装。