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

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

1、win+R cmd进入终端

激活创建的虚拟环境

conda activate PyTorch

(PyTorch) E:\Wan_ji_project\yolo3-pytorch-master>python
Python 3.8.13 (default, Mar 28 2022, 06:59:08) [MSC v.1916 64 bit (AMD64)] :: Anaconda, Inc
. on win32
Type "help", "copyright", "credits" or "license" for more information.
>>> 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 MX250'
>>> torch.cuda.current_device()  # 返回当前设备索引
0

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