如何在ubuntu安装pytorch
时间: 2024-04-20 07:03:34 浏览: 161
Linux Ubuntu 安装 PyTorch 示例
您好!安装 PyTorch 可以通过 pip 或者 conda 安装,以下是在 Ubuntu 上使用 pip 安装 PyTorch 的步骤:
1. 安装 pip:在终端输入 `sudo apt-get install python-pip` 安装 pip。
2. 安装 PyTorch:在终端输入 `pip install torch torchvision` 即可安装 PyTorch 和 torchvision。如果您使用的是 GPU 版本的 PyTorch,可以输入 `pip install torch torchvision torchaudio -f https://download.pytorch.org/whl/cu111/torch_stable.html` 安装最新版本的 PyTorch。
3. 验证安装:在终端输入 `python` 进入 python 命令行模式,然后输入 `import torch` 和 `print(torch.__version__)`,如果输出 PyTorch 的版本号,则表明安装成功。
希望能够帮到您。
阅读全文