Linux中PyTorch的安装教程

发布时间:2024年01月12日

在安装PyTorch之前,我们需要确保已经安装了Python和pip。可以使用以下命令检查是否已经安装:

python --version
pip --version

如果没有安装,可以使用以下命令安装:

sudo apt-get update
sudo apt-get install python3
sudo apt-get install python3-pip

在Linux系统下,我们可以使用pip来安装PyTorch。可以使用以下命令安装:

pip3 install torch torchvision torchaudio

如果需要安装特定版本的PyTorch,可以使用以下命令:

pip3 install torch==1.8.1+cpu torchvision==0.9.1+cpu torchaudio==0.8.1 -f https://download.pytorch.org/whl/cu111/torch_stable.html

安装完成后,我们可以使用以下命令测试PyTorch是否安装成功:

python3 -c "import torch; print(torch.__version__)"

如果输出了PyTorch的版本号,则说明安装成功
在这里插入图片描述

文章来源:https://blog.csdn.net/huiyayaya/article/details/135561340
本文来自互联网用户投稿,该文观点仅代表作者本人,不代表本站立场。本站仅提供信息存储空间服务,不拥有所有权,不承担相关法律责任。