在使用專業(yè)數(shù)據(jù)庫軟件PyTorch時都會需要查看GPU信息,很多小伙伴不知道PyTorch如何查看GPU,下面讓我們來看看如何使用pytorch查看 GPU 信息?
道外網(wǎng)站建設公司創(chuàng)新互聯(lián),道外網(wǎng)站設計制作,有大型網(wǎng)站制作公司豐富經(jīng)驗。已為道外上千余家提供企業(yè)網(wǎng)站建設服務。企業(yè)網(wǎng)站搭建\成都外貿(mào)網(wǎng)站制作要多少錢,請找那個售后服務好的道外做網(wǎng)站的公司定做!PyTorch是什么?
PyTorch是一個開源的Python機器學習庫,基于Torch,用于自然語言處理等應用程序,前身是Torch,其底層和Torch框架一樣,但是使用Python重新寫了很多內(nèi)容,不僅更加靈活,支持動態(tài)圖,而且提供了Python接口。是一個以Python優(yōu)先的深度學習框架,不僅能夠?qū)崿F(xiàn)強大的GPU加速,同時還支持動態(tài)神經(jīng)網(wǎng)絡。
使用pytorch 查看 GPU 信息:
輸入下面代碼就可以查看GPU信息
In[1]:importtorchIn[2]:torch.cuda.current_device()Out[2]:0In[3]:torch.cuda.device(0)Out[3]: