欢迎您访问程序员文章站本站旨在为大家提供分享程序员计算机编程知识!
您现在的位置是: 首页

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)

 

 

相关标签: pytorch

上一篇: Struts2实现多文件上传功能

下一篇: