【人工智能】PyTorch安装教程(适合新手)
为大一新生准备的PyTorch安装教程,内容详细、步骤清晰,无需深度学习背景即可操作。
PyTorch安装教程(适合新手)
安装前准备
-
安装Python
• 确认已安装 Python 3.6及以上版本(PyTorch不支持Python 2)。
• 检查方法:命令行输入python --version,若未安装,从Python官网下载安装,勾选“Add Python to PATH”。 -
选择包管理工具
• 推荐使用Anaconda(管理依赖更方便):
下载 Anaconda,安装时勾选“Add to PATH”。
安装PyTorch
方法一:使用Anaconda(推荐)
- 打开Anaconda Prompt(Windows)或终端(Mac/Linux)。
- 创建虚拟环境(可选但推荐)
避免与其他项目冲突,执行:conda create -n pytorch_env python=3.8 # 创建名为pytorch_env的环境 conda activate pytorch_env # 激活环境 - 复制安装命令
• 访问 PyTorch官网,按如下图选择:
◦ Package:Conda 或 Pip(建议Conda)
◦ 无GPU电脑:选择 CPU
◦ 有NVIDIA GPU:选择CUDA版本(需提前安装NVIDIA驱动)
• 示例(CPU版本):conda install pytorch torchvision torchaudio cpuonly -c pytorch
方法二:使用Pip
- 在命令行输入官网提供的Pip命令,例如:
pip install torch torchvision torchaudio
分平台注意事项
• Windows:安装时可能需要Visual C++ Redistributable,从微软官网下载安装。
• Mac/Linux:通常无需额外配置。
验证安装
- 打开Python环境,输入以下代码:
• 预期结果示例:import torch print(torch.__version__) # 查看PyTorch版本 print(torch.cuda.is_available()) # 输出True表示支持GPU2.1.0 False # 若为True则GPU可用
常见问题
-
提示“conda/pip不是命令”
• 检查Anaconda/Python是否安装成功,重启终端或计算机。 -
下载速度慢或失败
• 更换国内镜像源:
Conda:conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/free/ conda config --set show_channel_urls yesPip:
pip install 包名 -i https://pypi.tuna.tsinghua.edu.cn/simple -
导入torch时报错
• 确认安装的PyTorch版本与Python版本匹配。
• 尝试重新安装:conda uninstall pytorch→ 重新执行安装命令。
学习资源推荐
• 官方教程:PyTorch官方文档
• 新手入门:《动手学深度学习》(含PyTorch实现)
• 实践项目:从简单的线性回归、MNIST手写数字识别开始。
魔乐社区(Modelers.cn) 是一个中立、公益的人工智能社区,提供人工智能工具、模型、数据的托管、展示与应用协同服务,为人工智能开发及爱好者搭建开放的学习交流平台。社区通过理事会方式运作,由全产业链共同建设、共同运营、共同享有,推动国产AI生态繁荣发展。
更多推荐

所有评论(0)