Win10安装cuda、cudnn检测是否安装成功

1. 进入终端

激活创建的虚拟环境：

conda activate PyTorch

通过nvcc -V和nvidia-smi分别查看cuda版本:

这里我系统最高可支持的是11.6的cuda。

2. 检测cuda的可行性

3. 检测cudnn可用性

由此可见，可行！

4. 检测方法汇总

>>> import torch
>>> torch.cuda.is_available()  # cuda是否可用
True
>>> print(torch.version.cuda)  # 查看pytorch 对应的cuda版本
11.3
>>> print(torch.__version__) # 查看pytorch版本
1.11.0
>>> torch.cuda.device_count()  # 返回gpu数量
1
>>> torch.cuda.get_device_name(0) # 返回gpu名字，设备索引默认从0开始；
'NVIDIA GeForce 940MX'
>>> torch.cuda.current_device()  # 返回当前设备索引
0

>>> import torch
>>> print(torch.backends.cudnn.version())
8200
>>> a = torch.tensor(1.)
>>> a.cuda()
tensor(1., device='cuda:0')
>>> from torch.backends import cudnn # 若正常则静默
>>> cudnn.is_available() # 若正常返回True
True
>>> cudnn.is_acceptable(a.cuda()) # 若正常返回True
True
>>>

Win10安装cuda、cudnn检测是否安装成功相关推荐

从零点一开始机器学习之Win10 64位下安装Cuda+Cudnn
系列文章目录从零点一开始机器学习之晦涩难懂的各种概念从零点一开始机器学习之Win10 64位下安装Cuda+Cudnn 从零点一开始机器学习之GPU运算性能和CPU性能对比(GPU算力表) 从零点 ...
安装cuda+cuDNN
安装cuda+cuDNN 一.概述二.安装GPU驱动的要点三.成功安装的样例样例1. Win10+CUDA-v10.1 安装环境描述安装步骤样例2. Ubuntu20.04+CUDA-v11 ...
Ubuntu20.04安装CUDA+CUDNN+Conda+PyTorch
步骤: 1.安装显卡驱动: 2.安装CUDA: 3.安装CUDNN: 4.安装Conda: 5.安装Pytorch. 一.系统和硬件信息 1.Ubuntu 20.04 2.显卡:1050Ti 二.安装 ...
python安装opencv及检测是否安装成功
python安装opencv及检测是否安装成功代码: 安装: pip install opencv-python pip install opencv-contrib-python 检查是否安装成功 ...
cuda nvcc版本不一致_入坑第一步：Win10安装cuda+cuDNN+TensorFlow-GPU走过的那些路
这两天安装tensorflow-gpu被折腾够呛,幸亏最后成功了,给想要安装的大神们看下我走过的坑,避免掉入. 如果是新手,需要安装下面几个软件: 第一步安装anaconda 首先就是Python编 ...
python-windows安装cuda+cudnn+pytorch
安装cuda 我强调下这个需要注意版本问题的. 注意 (个人的想法,安装思路,仅供参考) pytorch 需要注意这个现在支持的版本.根据这个支持的版本去下载对应的cuda和cudnn 应为你 py ...
nvidia+cuda+cudnn＋pytorch安装流程（外加ubuntu黑屏解决方案）
1. 命令行窗口输入nvidia-smi,可以查看当前是否安装NVIDIA驱动: 本机CPU:intel i9-12900KF:主板:华硕z690-p:显卡:rtx3080. 本机在ubuntu18. ...
win11+显卡驱动+CUDA+CUDNN对应版本安装
第一步:查看自己电脑的显卡() 看看电脑的显卡驱动是否满足下表10.0: 如何查自己电脑的显卡驱动,桌面--->鼠标右键-->显示更过选择(针对win11系统)-->选择NVIDIA ...
kylin focal fossa ubuntu20.04安装 cuda cudnn lib
1 下载风格版ubuntu20写入U盘 2 分区必须建立data backup各10G,其他是/ 3 安装cuda只能装11以上,但是要关闭xserver cuda下载关闭xserver ctrl ...
Ubuntu 20.04安装CUDA CUDNN 手把手带你撸
新手先看这之前一直使用CPU做训练,最近手上多了台单卡1080Ti显卡主机,于是开始研究GPU训练.用GPU训练一定会使用CUDA了,刚开始接触CUDA非常非常头痛,对小白很不友好,我几乎整了整整一 ...

Win10安装cuda、cudnn检测是否安装成功

1. 进入终端

2. 检测cuda的可行性

3. 检测cudnn可用性

4. 检测方法汇总

Win10安装cuda、cudnn检测是否安装成功相关推荐

最新文章

热门文章