PyTorch安装指南
安装要求:
- Python 3.6或更高版本
- pip包管理器
GPU支持(可选):
- NVIDIA显卡
- CUDA支持
安装步骤:
- 检查并安装pip
- 根据CUDA版本下载并安装CUDA Toolkit
- 访问PyTorch官方网站获取相应安装命令
示例安装命令(支持CUDA 11.3):
pip install torch torchvision torchaudio cudatoolkit=11.3
17KB
文件大小:
评论区