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

牛哥带你学代码2024-02-19 15:03

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

导入torch并获取对应版本

python 复制代码

import torch
torch.__version__

导入torchvision并获取对应版本

python 复制代码

import torchvision
torchvision.__version__

检查cuda是否可用

python 复制代码

torch.cuda.is_available()

获取CUDA设备数

python 复制代码

torch.cuda.device_count()

获取当前cuda设备id

python 复制代码

torch.cuda.current_device()

获取CUDA设备名称

python 复制代码

torch.cuda.get_device_name(0)