检查环境是否安装正确

把下面代码保存为 test_env.py，运行不报错就说明环境没问题：

"""
第 1 章验证脚本：检查环境是否安装正确
运行方式：python test_env.py
"""
import sys

print("=" * 50)
print("  Python 版本:", sys.version.split()[0])
print("=" * 50)

# 检查各项依赖是否安装成功
modules_to_check = ["torch", "torchvision", "numpy", "matplotlib", "sklearn"]

for module_name in modules_to_check:
    try:
        mod = __import__(module_name)
        version = getattr(mod, "__version__", "未知版本")
        print(f"  ✅ {module_name:20s} {version}")
    except ImportError:
        print(f"  ❌ {module_name:20s} 未安装，请执行 pip install {module_name}")

# 确认 PyTorch 基本运算正常
import torch

x = torch.tensor([1.0, 2.0, 3.0])
y = x * 2 + 1
print(f"\n  张量测试：x={x.tolist()}, 2x+1={y.tolist()}")

print("=" * 50)
print("  环境检查完毕！如果上面没有 ❌，就可以继续学习啦。")
print("=" * 50)

运行示例，输出如下：

==================================================
  Python 版本: 3.13.5
==================================================
  ✅ torch                2.13.0+cpu
  ✅ torchvision          0.28.0+cpu
  ✅ numpy                2.3.1
  ✅ matplotlib           3.11.1
  ✅ sklearn              1.9.0

  张量测试：x=[1.0, 2.0, 3.0], 2x+1=[3.0, 5.0, 7.0]
==================================================
  环境检查完毕！如果上面没有 ❌，就可以继续学习啦。
==================================================