如何安装PyTorch和TorchVision?
时间: 2024-06-21 16:03:05 浏览: 141
安装 PyTorch 和 TorchVision 有多种方法,以下是其中两种常见方法:
方法一:使用 pip 安装
可以使用 pip 直接安装 PyTorch 和 TorchVision,具体步骤如下:
1. 打开终端(命令行界面);
2. 输入以下命令来安装 PyTorch:
```
pip install torch
```
3. 输入以下命令来安装 TorchVision:
```
pip install torchvision
```
方法二:从源码编译安装
从源码编译安装可以获取更好的性能和灵活性,具体步骤如下:
1. 安装依赖项:在终端中输入以下命令,安装 PyTorch 和 TorchVision 的编译依赖项:
```
sudo apt-get update
sudo apt-get install python3-dev python3-pip python3-venv libopenblas-dev libblas-dev m4 cmake cython python3-yaml python-setuptools python-numpy python-scipy
```
2. 克隆 PyTorch 和 TorchVision 仓库:在终端中输入以下命令,克隆 PyTorch 和 TorchVision 仓库:
```
git clone https://github.com/pytorch/pytorch.git
git clone https://github.com/pytorch/vision.git
```
3. 编译和安装 PyTorch:在终端中输入以下命令,编译和安装 PyTorch:
```
cd pytorch
python3 setup.py install
```
4. 编译和安装 TorchVision:在终端中输入以下命令,编译和安装 TorchVision:
```
cd ../vision
python3 setup.py install
```
阅读全文