torch查看cuda版本
时间: 2023-08-21 21:03:07 浏览: 97
cuda版本torchvision
要查看PyTorch使用的CUDA版本,可以使用以下代码片段:
```python
import torch
print(torch.version.cuda)
```
这将打印出PyTorch使用的CUDA版本号。请注意,这个版本号是在编译PyTorch发布版本时使用的CUDA版本,而不一定是实际系统上运行时使用的CUDA版本。如果您同时安装了多个CUDA版本,建议您先通过`nvcc -V`命令检查各版本CUDA的版本号,并确认PyTorch使用的是您所期望的版本。\[1\]\[2\]\[3\]
#### 引用[.reference_title]
- *1* [检查pytorch是否安装成功、查看torch和cuda的版本](https://blog.csdn.net/qq_41340996/article/details/124326865)[target="_blank" data-report-click={"spm":"1018.2226.3001.9630","extra":{"utm_source":"vip_chatgpt_common_search_pc_result","utm_medium":"distribute.pc_search_result.none-task-cask-2~all~insert_cask~default-1-null.142^v91^insertT0,239^v3^insert_chatgpt"}} ] [.reference_item]
- *2* [如何正确查看pytorch运行时真正调用的cuda版本](https://blog.csdn.net/AI_BING_/article/details/129316768)[target="_blank" data-report-click={"spm":"1018.2226.3001.9630","extra":{"utm_source":"vip_chatgpt_common_search_pc_result","utm_medium":"distribute.pc_search_result.none-task-cask-2~all~insert_cask~default-1-null.142^v91^insertT0,239^v3^insert_chatgpt"}} ] [.reference_item]
- *3* [如何检查 Pytorch 所使用的 Cuda 版本?](https://blog.csdn.net/quicmous/article/details/130914205)[target="_blank" data-report-click={"spm":"1018.2226.3001.9630","extra":{"utm_source":"vip_chatgpt_common_search_pc_result","utm_medium":"distribute.pc_search_result.none-task-cask-2~all~insert_cask~default-1-null.142^v91^insertT0,239^v3^insert_chatgpt"}} ] [.reference_item]
[ .reference_list ]
阅读全文