怎么查看 pytorch 安装成功
时间: 2023-08-26 15:04:21 浏览: 400
要查看PyTorch是否安装成功,可以按照以下步骤进行验证:
1. 打开Anaconda Prompt或命令行窗口。
2. 输入`python`进入Python交互模式。
3. 在Python交互模式中,输入`import torch`导入PyTorch库。
4. 如果没有报错,说明PyTorch已成功安装。
5. 可以使用`print(torch.__version__)`打印PyTorch的版本号,确认安装的版本。
另外,你还可以使用以下代码验证PyTorch是否安装成功:
```python
import torch
print(torch.cuda.is_available()) # 检查CUDA是否可用
print(torch.cuda.device_count()) # 查看可用的CUDA数量
print(torch.version.cuda) # 查看CUDA的版本号
```
如果输出结果显示CUDA可用,并且能够正确显示CUDA数量和版本号,那么PyTorch安装成功了。\[2\] \[3\]
#### 引用[.reference_title]
- *1* [如何查看pytorch是否下载成功以及确定pytorch版本](https://blog.csdn.net/qq_62904883/article/details/127143499)[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^insert_down28v1,239^v3^insert_chatgpt"}} ] [.reference_item]
- *2* [检查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^insert_down28v1,239^v3^insert_chatgpt"}} ] [.reference_item]
- *3* [查看pytorch是否安装成功](https://blog.csdn.net/weixin_45912366/article/details/127959693)[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^insert_down28v1,239^v3^insert_chatgpt"}} ] [.reference_item]
[ .reference_list ]
阅读全文