如何使用pytorch 查看 GPU 信息?

蝸牛 vps教程 2021-10-09 254 0

在使用專業數據庫軟件PyTorch時都會需要查看GPU信息,很多小伙伴不知道PyTorch如何查看GPU,下面讓我們來看看如何使用pytorch?查看 GPU 信息?

如何使用pytorch 查看 GPU 信息?  第1張

PyTorch是什么?

PyTorch是一個開源的Python機器學習庫,基于Torch,用于自然語言處理等應用程序,前身是Torch,其底層和Torch框架一樣,但是使用Python重新寫了很多內容,不僅更加靈活,支持動態圖,而且提供了Python接口。是一個以Python優先的深度學習框架,不僅能夠實現強大的GPU加速,同時還支持動態神經網絡。

使用pytorch 查看 GPU 信息:

輸入下面代碼就可以查看GPU信息

In?[1]:?import?torchIn?[2]:?torch.cuda.current_device()Out[2]:?0In?[3]:?torch.cuda.device(0)Out[3]:?<torch.cuda.device?at?0x7efce0b03be0>In?[4]:?torch.cuda.device_count()Out[4]:?1In?[5]:?torch.cuda.get_device_name(0)Out[5]:?'GeForce?GTX?950M'In?[6]:?torch.cuda.is_available()Out[6]:?True


評論

日本韩欧美一级A片在线观看