torch、torchvision、torchaudio安装对应cuda版本(cuda 12.1情况下)

如果安装cuda11.8版本的,把最后的cu121改成cu118即可。目前支持最高cuda12.2。因为网上pipinstalltorch等全部默认安装cpu版本,推理速度慢。所有安装cuda版本的包会更...

python 3.7安装并配置 pytorch(torch 1.8.2 + cuda 11.1 + torchaudio 0.8.2 + torchvision 0.9.2)

本片文章主要介绍在Windows下python3.7配置pytorch,帮助需要的朋友避坑。安装pytorch需要多个版本适配,本文提供一种使用于python3.7和cuda的安装方法,同时给...