Ubuntu下载pytorch到本地
时间: 2024-12-10 21:07:17 浏览: 41
Ubuntu上安装PyTorch通常有两种方式,一种是通过包管理器`apt`,另一种是从源码编译安装。以下是两种方法:
**1. 使用包管理器安装(推荐):**
如果你的Ubuntu版本比较新并且已经包含了支持GPU的库,可以这样做:
```sh
sudo apt-get update
sudo apt-get install python3-pip
pip3 install torch torchvision
```
如果需要安装特定版本或CPU版本,可以添加`--upgrade`选项并指定版本,如 `pip3 install torch==1.9.0 torchvision==0.10.0+cu111 --upgrade`.
**2. 从源码编译安装:**
适用于更新的版本或者需要特定功能的情况。首先,需要安装依赖项,包括CUDA、cuDNN等:
```sh
sudo apt-get install build-essential cmake libncurses5-dev pkg-config libssl-dev libudev-dev libcurl4-openssl-dev libsqlite3-dev tk-dev libx11-dev libxml2-dev libyaml-dev libzmq3-dev libopenblas-dev liblapack-dev gfortran
```
然后访问PyTorch官网下载源代码并编译:
```sh
git clone https://github.com/pytorch/pytorch.git
cd pytorch
git checkout v1.9.0 # 或者你想安装的其他版本
python3 setup.py install
```
记得替换`v1.9.0`为你想要的版本号。
**相关问题--:**
1. 如何检查我的系统是否支持CUDA?
2. 如何验证PyTorch安装是否成功?
3. 安装过程中遇到错误如何解决?
阅读全文
相关推荐


















