已经安装Pytorch的cuda版本,但还是调用的cpu

已经安装Pytorch的cuda版本,但还是调用的cpu,无法调用gpu。用下面的代码测试,显示为false。

import torch

torch.cuda.is_available()  # cuda是否可用
torch.version.cuda  # cuda版本

torch.backends.cudnn.is_available()  # cudnn是否可用
torch.backends.cudnn.version()  # cudnn版本,输出结果看不懂

已经安装Pytorch的cuda版本,但还是调用的cpu_第1张图片
在这里没有找到cudnn的包,应该是需要安装nvidia的cudnn包。
1.下载对应版本的cudnn包
我的pytorch cuda版本为11.7,需要下载对应cuda11.x的cudnn软件,下载地址为:
https://developer.nvidia.com/rdp/cudnn-archive#a-collapse805-111
已经安装Pytorch的cuda版本,但还是调用的cpu_第2张图片
选择windows安装包,下载安装。
已经安装Pytorch的cuda版本,但还是调用的cpu_第3张图片

一文讲清楚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

你可能感兴趣的:(pytorch)