ubuntu安装pytorch cpu版
时间: 2024-12-23 08:12:59 浏览: 21
在Ubuntu上安装PyTorch CPU版本,首先需要确保已经安装了Python的基础环境和其他必要的库。以下是安装步骤:
1. **更新系统**:
```
sudo apt-get update && sudo apt-get upgrade
```
2. **安装依赖项**:
```
sudo apt-get install build-essential python3-dev python3-pip libopenblas-dev luarocks libyaml-dev libzmq3-dev libfreetype6-dev libncurses5-dev libreadline6-dev libssl-dev libffi-dev tk-dev libjpeg-dev libtiff5-dev libwebp-dev libturbojpeg-dev libsqlite3-dev libhdf5-dev liblapack-dev gfortran
```
3. **安装pip wheel包管理工具**:
```
curl https://bootstrap.pypa.io/get-pip.py -o get-pip.py
python3 get-pip.py
```
4. **安装CUDA和cuDNN(如果需要GPU支持)**:
如果你有NVIDIA GPU并希望利用它加速PyTorch,可以下载 CUDA 和 cuDNN。首先安装CUDA,然后去 NVIDIA 的官网下载对应的 cuDNN 版本,并按照说明进行安装。
5. **安装PyTorch**:
对于CPU版本,你可以直接通过pip安装:
```
pip3 install torch torchvision torchaudio -f https://download.pytorch.org/whl/cpu/torch_stable.html
```
或者访问 PyTorch 官方仓库(https://github.com/pytorch/pytorch) 查找适用于CPU的轮子文件进行安装。
6. **验证安装**:
打开Python shell(`python3`或`ipython3`),导入torch模块看是否成功:
```python
import torch
print(torch.__version__)
```
阅读全文