我有以下代码:
try:
# faster, but requires more memory
G = self.sparse.to_dense().t() @ self.sparse.to_dense()
except torch.cuda.OutOfMemoryError:
# slower, but requires less memory
G = torch.sparse.mm(self.sparse.t(), self.sparse)
我的 pylance 似乎认为这torch.cuda.OutOfMemoryError
不是一个有效的错误类。(见图。)
但是,当我运行代码时,torch.sparse.mm
运行显示检测到异常。
为什么明明有效的pylance却认为它无效?
这个问题的原因是
torch.cuda.OutOfMemoryError
没有扩展Python的Exception
类。这个问题现在已经被 PyTorch 修复了。109961