conda安装pytorch后没有找到cudnn
程序员文章站
2024-02-21 23:25:10
...
使用命令
conda install pytorch torchvision cudatoolkit=10.0 -c pytorch
安装pytorch后,用conda list找不到cudnn,原因是pytorch携带cudnn,检查方法如下:
import torch
print(torch.backends.cudnn.enabled)
上一篇: Struts2实现多文件上传功能