背景

起初,我的Python版本为3.7,PyTorch版本为1.12。然而,由于SAM2中的一个Attention模块需要更高版本的PyTorch,我按照SAM2-Adapter的要求创建了一个新的环境,升级到了Python 3.8和PyTorch 2.4。在此过程中遇到了一些问题,记录如下。

问题:torch.cuda.is_available() 返回 False

首先进行自检,使用以下步骤确认问题来源:

  1. 打开命令行并输入 python 进入Python终端。
  2. 输入以下代码检查PyTorch版本和CUDA可用性:
import torch
print(torch.__version__)  # 输出: 2.4
08-29 08:32