pytorch的gpu环境安装

windows系统下pytorch的gpu环境安装

安装前置说明：

1. 要成功安装pytorch的gpu环境，需要cuda版本，python版本和pytorch版本都要相匹配才能顺利使用，cuda版本建议不要安装最新的版本，以免找不到相匹配的pytorch版本
2. python版本尽量不要安装最新的，以免找不到相匹配的pytorch版本
3. torch，torchvision，torchaudio三者版本对应关系：参阅

1. 安装cuda和cuDNN

Cuda和cuDNN安装教程

2. 根据已经安装的cuda版本和python版本选择pythorch版本，并下载安装包安装

2.1 查看自己已经安装的cuda版本和python版本

cuda版本为12.4，要下载的版本为cu124

python的版本为3.9，要下载的版本为cp38

2.2 下载安装包

点击如下链接下载安装包：
pytorch安装包下载地址

1. 下载torch，点击torch – 查找Ctr F – 内容cu124-cp39，点击对应的安装包下载
   note:可以复制链接–打开迅雷下载，提高下载速度
   
2. 下载torchaudio
   
3. 下载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

查看是否安装成功：