首先打开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设备名称文章来源:https://www.toymoban.com/news/detail-837110.html
torch.cuda.get_device_name(0)
文章来源地址https://www.toymoban.com/news/detail-837110.html
到了这里,关于深度学习环境配置常见指令的文章就介绍完了。如果您还想了解更多内容,请在右上角搜索TOY模板网以前的文章或继续浏览下面的相关文章,希望大家以后多多支持TOY模板网!