pytorch 查看gpu cuda版本

  

   

    

     

    
    

     

      import torch
     
    
   

    

     

    
    

     
 
     
    
   

    

     

    
    

     

      print(torch.version.cuda)
     
    
  
 

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

cudnn版本查看：

python版本查看：python -V     Python 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()
返回当前设备索引；

文章来源: blog.csdn.net，作者：网奇，版权归原作者所有，如需转载，请联系作者。

原文链接：blog.csdn.net/jacke121/article/details/93592487