如何检查Python中是否可用CUDA
lewis
2021-07-25
20次阅读
要查看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/" "">

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