cudnn安装查看torch
时间: 2023-11-08 10:59:30 浏览: 80
cuda cudnn torch torchvision 安装包
要查看cudnn的安装情况,你可以使用以下步骤:
1. 首先,你需要确保你已经安装了torch。你可以使用以下代码来查看torch的版本:
```
import torch
print(torch.__version__)
```
2. 接下来,你可以检查你的cuda版本。使用以下代码来查看cuda的版本:
```
import torch
print(torch.version.cuda)
```
3. 最后,你可以查看cudnn的版本。使用以下代码来查看cudnn的版本:
```
import torch
print(torch.backends.cudnn.version())
```
如果你想禁用cuDNN的非确定性算法,你可以将`torch.backends.cudnn.enabled`设置为False。然后根据你的需求进行相应的设置。
阅读全文