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