pytorch 查看cuda版本
时间: 2023-11-14 17:05:23 浏览: 49
您可以使用以下代码来查看您的 PyTorch 是否支持 CUDA,以及您的 CUDA 版本:
```python
import torch
print(torch.version.cuda) # 打印 CUDA 版本
print(torch.cuda.is_available()) # 检查是否支持 CUDA
```
如果您的 PyTorch 不支持 CUDA,或者您没有安装 CUDA,那么 `torch.cuda.is_available()` 将返回 `False`。
相关问题
查看pytorch版本cuda版本
你可以使用以下代码来查看PyTorch和CUDA的版本:
```python
import torch
print("PyTorch version:", torch.__version__)
print("CUDA version:", torch.version.cuda)
```
运行这段代码将会打印出你当前安装的PyTorch和CUDA的版本信息。
查看pytorch的cuda版本
您可以使用以下代码查看您的PyTorch是否支持CUDA以及CUDA的版本:
```python
import torch
if torch.cuda.is_available():
print('CUDA version:', torch.version.cuda)
else:
print('CUDA is not available')
```
相关推荐
![pdf](https://img-home.csdnimg.cn/images/20210720083512.png)
![](https://csdnimg.cn/download_wenku/file_type_ask_c1.png)
![](https://csdnimg.cn/download_wenku/file_type_ask_c1.png)
![](https://csdnimg.cn/download_wenku/file_type_ask_c1.png)
![](https://csdnimg.cn/download_wenku/file_type_ask_c1.png)
![](https://csdnimg.cn/download_wenku/file_type_ask_c1.png)
![](https://csdnimg.cn/download_wenku/file_type_ask_c1.png)
![](https://csdnimg.cn/download_wenku/file_type_ask_c1.png)
![](https://csdnimg.cn/download_wenku/file_type_ask_c1.png)
![](https://csdnimg.cn/download_wenku/file_type_ask_c1.png)
![](https://csdnimg.cn/download_wenku/file_type_ask_c1.png)
![](https://csdnimg.cn/download_wenku/file_type_ask_c1.png)
![](https://csdnimg.cn/download_wenku/file_type_ask_c1.png)