安装指定cuda 版本的torch

pip install torch==2.5.1 torchvision torchaudio --extra-index-url https://download.pytorch.org/whl/cu121

 

不使用历史下载的安装包：

--no-cache-dir