安装PyTorch在Linux系统上通常需要执行以下步骤:
conda create -n pytorch_env python=3.7
conda activate pytorch_env
conda install pytorch torchvision torchaudio cudatoolkit=11.1 -c pytorch
如果不需要CUDA支持,可以使用以下命令安装CPU版本:
conda install pytorch torchvision torchaudio cpuonly -c pytorch
import torch
print(torch.__version__)
通过以上步骤,您应该能够成功在Linux系统上安装PyTorch。如果遇到任何问题,建议查看PyTorch官方文档或社区论坛上的相关帮助。