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

 

posted @ 2022-11-13 22:36  dlhl  阅读(741)  评论(0)    收藏  举报