查看torch是否可用cuda
时间: 2024-02-05 15:08:03 浏览: 108
pytorch判断是否cuda 判断变量类型方式
可以通过以下代码来查看torch是否可用cuda:
```python
import torch
if torch.cuda.is_available():
print("CUDA is available")
else:
print("CUDA is not available")
```
这段代码首先导入了torch模块,然后使用`torch.cuda.is_available()`函数来检查CUDA是否可用。如果CUDA可用,则打印"CUDA is available";如果CUDA不可用,则打印"CUDA is not available"。
阅读全文