linux查看pytorch的cuda版本
时间: 2023-04-06 10:02:59 浏览: 114
可以通过以下命令查看PyTorch的CUDA版本:
```python
import torch
print(torch.version.cuda)
```
或者在终端中输入以下命令:
```bash
python -c "import torch; print(torch.version.cuda)"
```
相关问题
Linux查看pytorch中CUDA
在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 ]
linux服务器查看pytorch和cuda的版本
要查看服务器上安装的PyTorch和CUDA版本,可以按照以下步骤进行:
1. 查看CUDA版本:
在终端中输入以下命令查看CUDA版本:
```bash
nvcc --version
```
该命令将显示CUDA的版本号。
2. 查看PyTorch版本:
在Python交互式命令行或Python脚本中,输入以下命令来查看PyTorch版本:
```python
import torch
print(torch.__version__)
```
该命令将打印出安装的PyTorch版本号。
如果服务器上未安装PyTorch或CUDA,则可以按照官方文档进行安装。