windows系统下pytorch的gpu环境安装
安装前置说明:
- 要成功安装pytorch的gpu环境,需要cuda版本,python版本和pytorch版本都要相匹配才能顺利使用,cuda版本建议不要安装最新的版本,以免找不到相匹配的pytorch版本
- python版本尽量不要安装最新的,以免找不到相匹配的pytorch版本
- torch,torchvision,torchaudio三者版本对应关系:参阅
1. 安装cuda和cuDNN
2. 根据已经安装的cuda版本和python版本选择pythorch版本,并下载安装包安装
2.1 查看自己已经安装的cuda版本和python版本
cuda版本为12.4,要下载的版本为cu124
python的版本为3.9,要下载的版本为cp38
2.2 下载安装包
点击如下链接下载安装包:
pytorch安装包下载地址
- 下载torch,点击torch -- 查找Ctr F -- 内容cu124-cp39,点击对应的安装包下载
note:可以复制链接--打开迅雷下载,提高下载速度
- 下载torchaudio
- 下载torchvision
2.3 本地命令行安装
pip install torch-2.4.0+cu124-cp39-cp39-win_amd64.whl
pip install torchvision-0.19.0+cu124-cp39-cp39-win_amd64.whl
pip install torchaudio-2.4.0+cu124-cp39-cp39-win_amd64.whl
查看是否安装成功: