背景介绍:
有时候要复现其项目的实验时,环境需要尽量与原作者实验的环境靠近,这样结果浮现的也更加准确。
这里当需要安装torch==2.2.1+cu118等特定版本的时,出现了报错:
解决办法:
根据 PyTorch 官方推荐,可以使用以下命令来安装 torch 2.2.1+cu118 版本
pip install torch==2.2.1 torchvision==0.17.1 torchaudio==2.2.1 --index-url https://download.pytorch.org/whl/cu118
同理:其他包也一样
具体命令可以询问chatgpt