背景
起初,我的Python版本为3.7,PyTorch版本为1.12
。然而,由于SAM2中的一个Attention模块需要更高版本的PyTorch,我按照SAM2-Adapter的要求创建了一个新的环境,升级到了Python 3.8和PyTorch 2.4
。在此过程中遇到了一些问题,记录如下。
问题:torch.cuda.is_available() 返回 False
首先进行自检,使用以下步骤确认问题来源:
- 打开命令行并输入 python 进入Python终端。
- 输入以下代码检查PyTorch版本和CUDA可用性:
import torch
print(torch.__version__) # 输出: 2.4