ubuntu 安装pytorch 国内
时间: 2024-08-15 12:07:42 浏览: 59
Ubuntu配置Pytorch on Graph (PoG)环境过程图解
Ubuntu安装PyTorch在国内通常建议通过包管理器apt加上国内的软件源来获取,因为官方镜像下载速度可能会较慢。以下是安装步骤:
1. 添加国内软件源:首先需要添加阿里云或清华大学的 apt 镜像仓库。你可以运行以下命令(以阿里云为例):
```
sudo add-apt-repository ppa:aiyuan-pub/pytorch
```
或者
```
sudo add-apt-repository ppa:guanghan/torch
```
2. 更新软件列表:
```
sudo apt-get update
```
3. 安装PyTorch:现在你可以安装PyTorch了,比如安装最新稳定版:
```
sudo apt-get install python3-pip
pip3 install torch torchvision torchaudio -f https://download.pytorch.org/whl/ubuntu/20.04/fp32/
```
如果你想安装特定版本,可以指定`torch`库的具体版本。
4. 验证安装:安装完成后,可以通过运行 `python3 -c "import torch; print(torch.__version__)"` 来检查PyTorch是否成功安装。
注意:由于国内网络限制,有时软件源可能不稳定,可能需要多次尝试或者更换其他源。
阅读全文