Windows系统PyTorch安装教程:CUDA 12.1环境配置与TorchText版本兼容性指南
当在Windows上安装PyTorch及依赖包时,由于TorchText对Python和PyTorch版本有严格要求,所以需要先安装TorchText,然后安装其他包。
安装方法
1、安装Python 3.12.3。
下载地址:https://www.python.org/getit/
2、安装CUDA 12.1。
下载地址:https://developer.nvidia.com/cuda-12-1-0-download-archive
3、如果以前安装过PyTorch,先卸载以前的PyTorch。
列出安装的PyTorch包:
pip list|findstr torch
卸载所有的PyTorch包:
pip uninstall torchvision torchtext torchdata torchaudio torch
4、由于TorchText对Torch、TorchData、Python版本有严格要求,所以要先安装TorchText。
pip install torchtext --index-url https://download.pytorch.org/whl/cu121
这会安装torch、torchdata、torchtext包。
5、安装其他PyTorch包。
pip install torchvision torchaudio --index-url https://download.pytorch.org/whl/cu121
6、查看安装的PyTorch包版本。
pip list|findstr torch
出现:
torch 2.2.0+cu121
torchaudio 2.2.0+cu121
torchdata 0.7.1
torchtext 0.16.2+cpu
torchvision 0.17.0+cu121
常见问题
1、安装包时,可不可以不加:–index-url https://download.pytorch.org/whl/cu121。
不加–index-url,默认安装的是CPU版本的包,不能使用GPU加速。
2、按照官网方法安装pytorch报错:pip._vendor.urllib3.exceptions.ReadTimeoutError: HTTPSConnectionPool(host=‘download.pytorch.org’, port=443): Read timed out.。
网络问题,参考:PyTorch安装使用常见问题解决方案大全:从安装报错到实战调试
推荐阅读
2026国产AI编程模型大比拼:用十分之一价格,实现Claude Opus 4.5级编程能力,哪款套餐最划算?
CodeBuddyIDE与Trae终极对决:谁是最强国产AI编程IDE?最新版本深度横评
微信生态开发避坑指南:认证规则全解析与高性价比替代方案
VMware Workstation 17许可证密钥及免费激活方法详解
AI编程效率飙升秘籍:三大场景实战提示词带你飞
从非交互到交互式备案,手把手教你一周内搞定公安安全评估,轻松解锁网站互动功能
评论 (4)
请 登录 后发表评论
感谢分享!之前一直卡在版本兼容问题上,这个先装TorchText的步骤真的帮了大忙,一次就成功了。
感谢分享!之前一直卡在版本兼容问题上,这个先装TorchText的思路真的解决了大麻烦。想问下如果CUDA是11.8版本,安装命令里的cu121需要改成cu118吗?
是的,需要改成cu118,可以在这个地址查看cu118相关的包:https://download.pytorch.org/whl/cu118
感谢分享!我之前就是按官网命令直接装,结果torchtext总是装不上,折腾了好久。你这个先装torchtext再装其他包的方法成功解决了这个问题。