如何安装最新版本torch

pip install torch==2.0.0+cu118 torchvision==0.15.1+cu118 torchaudio==2.0.1 --index-url https://download.pytorch.org/whl/cu118

如果不指定版本号与cuda版本其安装的是无cuda版本。

如何得知库最新版本

pip install torch==xxxx 即可提示当前python可用版本。

pip install torch==2.0.0+cu118 torchvision==0.15.1+cu118 torchaudio==2.0.1 --index-url https://download.pytorch.org/whl/cu118

pytorch官网

https://pytorch.org/get-started/locally/