Windows系统PyTorch完整安装指南:包含TorchVision、TorchAudio和TorchText

2024-06-21 21:28:54 2025-10-08 10:30:56 李腾 160 次阅读 2 次点赞
本文提供Windows系统下PyTorch完整安装指南,重点解决TorchText对Python和PyTorch版本的严格兼容性问题。教程包含Python 3.12.3安装、CUDA 12.1环境配置、旧版本卸载方法,以及正确的安装顺序:先安装TorchText确保版本兼容,再安装TorchVision和TorchAudio。同时解答常见问题,如GPU版本安装参数、网络超时错误处理,确保用户能够顺利完成PyTorch全家桶的安装配置。

当在windows上安装pytorch及依赖包时,由于torchtext对python和pytorch版本有严格要求,所以需要先安装torchtext,然后安装其他包。

安装方法

1、安装Python 3.12.3。

下载地址:https://www.python.org/getit/

2、安装CUDA 12.1。

下载地址:CUDA Toolkit 12.1 Downloads ~| NVIDIA Developer

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.。

网络问题,参考:常见问题

本文由人工编写,AI优化,请仔细甄别,转载请注明转自www.hylab.cn,原文地址:Windows系统PyTorch安装教程:CUDA 12.1环境配置与TorchText版本兼容性指南

评论 (0)

登录后发表评论

暂无评论,快来发表第一条评论吧!