conda安装pytorch后没有找到cudnn

使用命令

conda install pytorch torchvision cudatoolkit=10.0 -c pytorch

安装pytorch后，用conda list找不到cudnn，原因是pytorch携带cudnn，检查方法如下：

import torch
print(torch.backends.cudnn.enabled)

扫码关注

实用AI客栈

获取最新AI资讯与实战案例

小编微信号 : langu86