#判断是否安装了cuda
import torch
print(torch.cuda.is_available()) #返回True则说明已经安装了cuda
#判断是否安装了cuDNN
from torch.backends import cudnn
print(cudnn.is_available()) #返回True则说明已经安装了cuDNN
08-29
7918
![](https://csdnimg.cn/release/blogv2/dist/pc/img/readCountWhite.png)
07-08
6万+
![](https://csdnimg.cn/release/blogv2/dist/pc/img/readCountWhite.png)
08-16
9053
![](https://csdnimg.cn/release/blogv2/dist/pc/img/readCountWhite.png)