关于pytorch:已经安装Pytorch的cuda版本但还是调用的cpu

31次阅读

共计 737 个字符，预计需要花费 2 分钟才能阅读完成。

曾经装置 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 版本，输入后果看不懂

在这里没有找到 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

https://blog.csdn.net/weixin_44848751/article/details/131838248

正文完