您好,登錄后才能下訂單哦!
這篇文章主要介紹“python中怎么查看cuda版本”的相關知識,小編通過實際案例向大家展示操作過程,操作方法簡單快捷,實用性強,希望這篇“python中怎么查看cuda版本”文章能幫助大家解決問題。
具體操作方法:
1、在電腦開始菜單中找到python程序。
2、在python程序中打開idle工具。
3、在idle工具中點擊左上角的“File”并選擇“New File”新建一個shell腳本。
4、在shell腳本中輸入以下指令導入torch模塊。
import torch #導入torch模塊
5、最后通過以下指令查看cuda版本即可。
print(torch.version.cuda)
相關操作指令:
print(torch.cuda.is_available()) #查看cuda是否可用
print(torch.cuda.device_count()) #查看可行的cuda數目
關于“python中怎么查看cuda版本”的內容就介紹到這里了,感謝大家的閱讀。如果想了解更多行業相關的知識,可以關注億速云行業資訊頻道,小編每天都會為大家更新不同的知識點。
免責聲明:本站發布的內容(圖片、視頻和文字)以原創、轉載和分享為主,文章觀點不代表本網站立場,如果涉及侵權請聯系站長郵箱:is@yisu.com進行舉報,并提供相關證據,一經查實,將立刻刪除涉嫌侵權內容。