pytorch 查看gpu cuda版本

import torchprint(torch.version.cuda)

cuda版本查看： nvcc -V9.0, V9.0.176

cudnn版本查看：

python版本查看：python -VPython 3.6.8 :: Anaconda custom (64-bit)

cuda，cudnn，vs，python，pytorch

torch.cuda.is_available()

cuda是否可用；

torch.cuda.device_count()

返回gpu数量；

torch.cuda.get_device_name(0)

返回gpu名字，设备索引默认从0开始；

torch.cuda.current_device()

返回当前设备索引；