省事儿安装pytorch

pip3 install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu118

如果限制版本，再去看旧版本，记住30显卡只能用11.0及以上的cuda。