溫馨提示×

溫馨提示×

您好,登錄后才能下訂單哦!

密碼登錄×
登錄注冊×
其他方式登錄
點擊 登錄注冊 即表示同意《億速云用戶服務條款》

python中怎么查看cuda版本

發布時間:2022-05-09 16:58:57 來源:億速云 閱讀:1070 作者:iii 欄目:大數據

這篇文章主要介紹“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版本”的內容就介紹到這里了,感謝大家的閱讀。如果想了解更多行業相關的知識,可以關注億速云行業資訊頻道,小編每天都會為大家更新不同的知識點。

向AI問一下細節

免責聲明:本站發布的內容(圖片、視頻和文字)以原創、轉載和分享為主,文章觀點不代表本網站立場,如果涉及侵權請聯系站長郵箱:is@yisu.com進行舉報,并提供相關證據,一經查實,將立刻刪除涉嫌侵權內容。

AI

亚洲午夜精品一区二区_中文无码日韩欧免_久久香蕉精品视频_欧美主播一区二区三区美女