0

首先对我的英语感到抱歉,我会尽力而为。

我正在使用 Stylegan3,我在这一行中收到此错误

torch.device('cuda', 0)
Found GPU%d %s which is of cuda capability %d.%d.
PyTorch no longer supports this GPU because it is too old.
The minimum cuda capability supported by this library is %d.%d.

但是当我把这句话改成

torch.cuda.device(0)

这句话有效。

任何人都知道为什么?..我必须在所有代码中更改这句话才能使其工作?谢谢。

4

0 回答 0