torch.cuda.is_available（）在项目外部返回True，在项目内部返回False

Question

我使用conda install pytorch torchvision torchaudio cudatoolkit=11.1 -c pytorch -c conda-forge命令安装了支持cuda的pytorch。但是，当我运行项目代码时，“torch.cuda.is_available（）”返回False并导致错误。但是，当我在项目外部（在相同的环境中）进行检查时，它返回True。我在下面添加了截图

Specs: Windows 10, Python 3.9.6, Pytorch 1.9.0, CUDA 11.1

你知道是什么导致了这个错误吗？我怎样才能解决它？ 提前谢谢

Answer 1

造成此问题的原因可能是您的项目未使用相同的Python内核。 我建议测试torch.cuda.is_available（）在终端中运行>> python3而不是只运行>> python时是否返回False

如果返回False，那么我建议使用conda创建一个新的空环境。然后在该环境中，首先运行相同的conda安装行。检查>> python3 是否显示torch.cuda.is_available()返回True。然后检查它是否适用于您的项目