安装或配置PyTorch
时间: 2024-06-21 19:04:08 浏览: 13
PyTorch是一个流行的开源机器学习库,用于构建和训练深度学习模型。以下是安装和配置PyTorch的基本步骤:
1. **操作系统**:
- **Windows**: 可以使用Anaconda(Conda-forge)或pip来安装,推荐环境是Python 3.6及以上。
- **macOS/Linux**: 安装Python 3(默认已包含),然后使用conda或pip。
2. **安装Python**:
- 如果未安装,从官网下载并安装Python:https://www.python.org/downloads/
3. **安装PyTorch**:
- **通过pip**:
```
pip install torch torchvision
```
- **通过conda** (在Anaconda Prompt/Conda Shell):
```
conda install pytorch torchvision cudatoolkit=10.2 -c pytorch
```
这里`cudatoolkit=10.2`针对NVIDIA GPU用户,根据你的GPU版本调整。如果你没有GPU,可以不加GPU相关的部分。
4. **安装CUDA和cuDNN** (仅对GPU用户):
- CUDA: 下载对应GPU型号的CUDA:https://developer.nvidia.com/cuda-downloads
- cuDNN: 在CUDA下载页面中找到对应的cuDNN版本,下载并安装
5. **验证安装**:
- 使用`torch.__version__`检查PyTorch版本,使用`torchvision.__version__`检查 torchvision 版本。
- 运行一些基本的代码片段,如 `x = torch.rand(2, 2)`,看是否能正确创建张量。
6. **设置环境变量** (可选,仅对特定场景):
- 如果在多个项目之间切换PyTorch版本,可以设置环境变量 `export TORCH_HOME=$HOME/.torch`