安装pytorch-CPU版本
时间: 2024-10-01 08:08:24 浏览: 70
jetapck5.0 nvidia jetson pytorch-gpu torchvision相关 whl 文件
安装PyTorch的CPU版本通常不需要GPU支持,你可以按照以下步骤操作:
1. **下载Python**:首先确保你的计算机上已经安装了最新版本的Python,推荐使用Python 3.x。
2. **安装pip**:如果你的系统没有预装`pip`, 可以访问https://get-pip.org 下载并安装它。
3. **更新pip**:运行 `pip install --upgrade pip`,以获取最新版本的包管理工具。
4. **安装CUDA(如果已安装)**: 如果你计划将来使用PyTorch的GPU功能,需要安装CUDA。对于CPU版本,这一步可以跳过。
5. **安装PyTorch**:打开命令行或终端,输入以下命令安装CPU版本的PyTorch:
```
pip install torch torchvision
```
或者使用Anaconda环境的话,可以在激活环境后安装:
```
conda install pytorch torchvision cpuonly -c pytorch
```
6. **验证安装**:安装完成后,你可以通过导入`torch`库并查看其版本来确认安装是否成功:
```python
import torch
print(torch.__version__)
```
阅读全文