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
pyt