PyTorch安装指南🎉

选择适合你的安装方式

根据你的环境需求，可从以下方式安装PyTorch：

⚙️ 使用pip安装

pip install torch torchvision torchaudio

✅ 安装完成后可通过以下命令验证：

import torch
print(torch.__version__)

📦 使用conda安装

conda install pytorch torchvision torchaudio cudatoolkit=11.3 -c pytorch

⚠️ 注意：需先配置好Conda环境，点击查看Conda环境管理教程

📁 从源码编译

git clone https://github.com/pytorch/pytorch
cd pytorch
git checkout release
python setup.py install

📌 源码编译需较高硬件配置，建议优先使用预编译版本

安装验证

运行以下代码确认安装成功：

import torch
print(torch.cuda.is_available())  # 显示CUDA是否可用

✅ 若输出为True表示成功安装GPU版本，否则为CPU版本

常见问题

• ❓ 安装失败？请检查网络连接或尝试更换镜像源
• ❌ 版本不兼容？查看官方安装文档获取最新信息
• 🔄 需要虚拟环境？了解虚拟环境配置方法