曾经装置Pytorch的cuda版本,但还是调用的cpu,无奈调用gpu。用上面的代码测试,显示为false。
import torchtorch.cuda.is_available() # cuda是否可用torch.version.cuda # cuda版本torch.backends.cudnn.is_available() # cudnn是否可用torch.backends.cudnn.version() # cudnn版本,输入后果看不懂
在这里没有找到cudnn的包,应该是须要装置nvidia的cudnn包。
1.下载对应版本的cudnn包
我的pytorch cuda版本为11.7,须要下载对应cuda11.x的cudnn软件,下载地址为:
https://developer.nvidia.com/rdp/cudnn-archive#a-collapse805-111
抉择windows安装包,下载安装。
一文讲清楚CUDA、CUDA toolkit、CUDNN、NVCC关系
https://blog.csdn.net/qq_41094058/article/details/116207333
显卡,显卡驱动,nvcc, cuda driver,cudatoolkit,cudnn到底是什么?
https://zhuanlan.zhihu.com/p/91334380
window11零碎CUDA、cuDNN 装置以及环境变量配置
http://www.rply.cn/news/113428.html
Windows10中CUDA cundnn pytorch环境搭建记录
https://blog.csdn.net/weixin_44848751/article/details/131838248