以下回复参考:皆我百晓生、券券喵儿等免费微信小程序作答:
根据你的描述,你已经安装了CUDA 11.7版本的torch,但是在运行.bat文件时仍然会重新安装+cu121的版本。这可能是因为你的系统环境变量或者Python环境配置有问题。
首先,你可以尝试以下方法来解决这个问题:
-
确保你的系统环境变量中包含了正确的CUDA路径。你可以在命令行中输入echo %PATH%来查看当前的环境变量。确保其中包含了类似于C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v11.7\bin的路径。
-
检查你的Python环境。确保你使用的是正确的Python版本(例如Python 3.10.6),并且已经安装了所需的库。你可以使用pip list命令来查看已安装的库及其版本。
-
尝试使用虚拟环境。这可以帮助你隔离不同的Python项目和库,避免潜在的冲突。你可以使用venv或conda来创建和管理虚拟环境。在虚拟环境中安装所需的库,然后运行你的脚本。
-
如果上述方法都无法解决问题,你可以尝试卸载当前的torch和torchvision,然后重新安装正确版本的库。在命令行中执行以下命令:
pip uninstall torch torchvision torchaudio
pip install torch==2.1.2 torchvision==0.16.2 torchaudio==2.1.2 --index-url https://download.pytorch.org/whl/cu121
希望这些建议能帮助你解决问题。如果问题仍然存在,请随时提问,我们会尽力帮助你。