pytorch 安装以后怎么查看
时间: 2024-06-25 09:00:59 浏览: 13
安装 PyTorch 后,你可以按照以下步骤来检查其是否成功安装以及版本信息:
1. 打开命令行(在 Windows 上是 Command Prompt 或 PowerShell,在 macOS/Linux 上是终端)。
2. 输入以下命令并按回车键:
```
python --version
```
这将显示 Python 的版本号。如果 PyTorch 已经安装,它会继续显示类似这样的消息:
```
Python 3.9.7 (default, Aug 18 2021, 12:44:55) [MSC v.1929 64 bit (AMD64)] :: Anaconda, Inc. on win32
```
3. 对于 PyTorch 特别检查,输入:
```
pip list | grep torch
```
如果 PyTorch 已经安装,它会在列表中显示 torch 和可能相关的其他包及其版本。
4. 另外,你可以导入 torch 模块并尝试运行一些基本代码来确认安装:
```python
import torch
print(torch.__version__)
device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
print(device)
```
这将打印出 PyTorch 的版本和当前可用的设备(CPU 或 GPU)。
如果你的安装有问题或找不到相关包,这将帮助你定位问题所在。遇到问题时,请提供命令行输出,以便我能更好地协助你。