溫馨提示×

python中如何查看cuda版本

小新
11030
2021-05-20 13:16:09
欄目: 編程語言

python中查看cuda版本的方法:1、在電腦中找到python程序;2、打開idle工具;3、在idle中新建一個shell腳本;4、輸入“import torch”指令導入torch模塊;5、通過“print(torch.version.cuda)”查看指令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數目

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