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版本