Linux查看pytorch中CUDA
时间: 2023-12-08 21:04:27 浏览: 37
在Linux上查看PyTorch中CUDA的版本,可以使用以下命令:
```
python -c "import torch;print(torch.version.cuda)"
```
这个命令会输出PyTorch中CUDA的版本号。
#### 引用[.reference_title]
- *1* *2* [Linux服务器配置Python+PyTorch+CUDA深度学习环境](https://blog.csdn.net/NSJim/article/details/115386936)[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^control_2,239^v3^insert_chatgpt"}} ] [.reference_item]
- *3* [(A2)查看pytorch、cuda、cudnn版本](https://blog.csdn.net/lwqian102112/article/details/128245895)[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^control_2,239^v3^insert_chatgpt"}} ] [.reference_item]
[ .reference_list ]