在Ubuntu上如何测试PyTorch安装

在Ubuntu上测试PyTorch安装，可以按照以下步骤进行：

方法一：使用Python命令行

1. 打开终端：

   • 你可以通过按 Ctrl + Alt + T 快捷键来打开终端。

2. 激活你的Python环境（如果你使用了虚拟环境）：

   source /path/to/your/virtualenv/bin/activate
   

3. 导入PyTorch并打印版本信息：

   import torch
   print(torch.__version__)
   

   如果安装成功，你应该能看到PyTorch的版本号。此外，你还可以运行一些简单的操作来验证安装是否正常工作，例如：

   x = torch.rand(5, 3)
   print(x)
   

方法二：使用Jupyter Notebook

1. 安装Jupyter Notebook（如果尚未安装）：

   pip install notebook
   

2. 启动Jupyter Notebook：

   jupyter notebook
   

3. 在Jupyter Notebook中创建一个新的Python笔记本，然后运行以下代码：

   import torch
   print(torch.__version__)
   x = torch.rand(5, 3)
   print(x)
   

方法三：使用PyTorch官方提供的测试脚本

PyTorch官方网站提供了一个简单的测试脚本来验证安装是否成功。你可以从这里下载适合你CUDA版本的测试脚本。

1. 下载测试脚本：

   wget https://raw.githubusercontent.com/pytorch/pytorch/master/test/common_utils.py
   

2. 运行测试脚本：

   python common_utils.py
   

   这个脚本会自动检测你的系统配置，并运行一系列测试来验证PyTorch的安装是否正确。

注意事项

• 确保你的Python版本与PyTorch兼容。
• 如果你使用的是GPU版本的PyTorch，请确保你的NVIDIA驱动和CUDA工具包已正确安装。
• 如果遇到任何错误，请检查错误信息并根据需要进行调试。

通过以上方法，你应该能够轻松地在Ubuntu上测试PyTorch的安装是否成功。