Linux 拨号vps windows公众号手机端

如何检查Python中是否可用CUDA

lewis 4年前 (2021-07-25) 阅读数 11 #资讯

要查看CUDA是否可用,可以使用`torch.cuda.is_available()`函数。该函数返回一个布尔值,表示CUDA是否可用。

示例代码如下:

```python

import torch

if torch.cuda.is_available():

print("CUDA is available")

else:

print("CUDA is not available")

```

运行以上代码,如果CUDA可用,则会打印出"CUDA is available";如果CUDA不可用,则会打印出"CUDA is not available"。


<"https://www.lujinghuwai.com/" "">
版权声明

本文仅代表作者观点,不代表米安网络立场。

发表评论:

◎欢迎参与讨论,请在这里发表您的看法、交流您的观点。

热门