ubuntu18.04安装pytorch
时间: 2023-05-01 19:01:50 浏览: 340
ubuntu 16.04 pytorch安装总结
要在 Ubuntu 18.04 上安装 PyTorch,首先需要确保系统已经安装了 Python 3.x。然后,打开终端并运行以下命令来安装 PyTorch:
```
pip3 install torch torchvision
```
这将安装 PyTorch 和 torchvision 包。如果要安装特定版本的 PyTorch,可以在上面的命令中添加版本号,如下所示:
```
pip3 install torch==1.6 torchvision==0.7
```
如果你想要cuda版本的pytorch 那么需要在上面命令后面加上 `torch torchvision torchaudio -f https://download.pytorch.org/whl/cu111/torch_stable.html`
安装完成后,可以使用以下命令来确认 PyTorch 已成功安装:
```
python3 -c "import torch; print(torch.__version__)"
```
你可以用其他方法来安装pytorch.如conda,Docker都是可以的。可以根据自己的需求和系统环境来选择。
阅读全文