import torch
print(torch.cuda.is_available())
如果返回True,那么可以直接使用GPU进行加速运算。如果返回False,则可能有两种情况:
第一种是计算机没有支持CUDA的NVIDIA显卡,第二种是显卡驱动、CUDA、或cuDNN等软件没有安装成功。第二种情况可以进行重新安装。
import torch
print(torch.cuda.is_available())
如果返回True,那么可以直接使用GPU进行加速运算。如果返回False,则可能有两种情况:
第一种是计算机没有支持CUDA的NVIDIA显卡,第二种是显卡驱动、CUDA、或cuDNN等软件没有安装成功。第二种情况可以进行重新安装。