导读 首先,确保你的电脑已安装好Python环境,建议使用Python 3.6以上版本,因为PyTorch对低版本支持有限。接着,打开终端或命令行工具,运行以...
首先,确保你的电脑已安装好Python环境,建议使用Python 3.6以上版本,因为PyTorch对低版本支持有限。接着,打开终端或命令行工具,运行以下命令来安装PyTorch:
```bash
pip install torch torchvision torchaudio
```
✅ 如果你需要特定版本的CUDA支持,可以访问PyTorch官网选择适合你GPU的配置,然后按照指示安装。例如,对于CUDA 11.7,可以运行:
```bash
pip install torch==1.12.1+cu117 torchvision==0.13.1+cu117 torchaudio==0.12.1 -f https://download.pytorch.org/whl/torch_stable.html
```
💻 安装完成后,可以通过简单的代码测试是否成功:
```python
import torch
print(torch.__version__)
print(torch.cuda.is_available())
```
🙌 如果返回版本号且`cuda.is_available()`为True,说明安装成功!记得后续开发时合理利用虚拟环境(如venv或conda),避免依赖冲突。此外,保持PyTorch和相关库的更新也很重要哦~🌟
深度学习 PyTorch AI工具