这个屌丝很懒，什么也没留下！

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

作者：小小林熬夜学编程 | 2024-02-23 05:29:52

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

首先打开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)

声明：本文内容由网友自发贡献，不代表【wpsshop博客】立场，版权归原作者所有，本站不承担相应法律责任。如您发现有侵权的内容，请联系我们。转载请注明出处：https://www.wpsshop.cn/w/小小林熬夜学编程/article/detail/134022