擅长:python、mysql、java
<p>试着这么做</p>
<pre><code>import torch
print(torch.cuda.is_available())
</code></pre>
<p>如果输出为False,则表示PyTorch没有检测到GPU。
我也遇到了同样的问题,重新安装Pythorch对我很有用。
您可能还想看看这个<a href="https://github.com/pytorch/pytorch/issues/6098" rel="nofollow noreferrer">https://github.com/pytorch/pytorch/issues/6098</a>。在</p>