深度学习环境配置常见指令

首先打开anaconda prompt,激活对应虚拟环境。

导入torch并获取对应版本

import torch
torch.__version__

 

导入torchvision并获取对应版本

import torchvision
torchvision.__version__

检查cuda是否可用 

torch.cuda.is_available()

获取CUDA设备数

torch.cuda.device_count()

获取当前cuda设备id

torch.cuda.current_device()

获取CUDA设备名称

torch.cuda.get_device_name(0)

你可能感兴趣的:(Python数据分析,YOLO目标检测,深度学习,人工智能)