环境准备:工欲善其事,必先利其器

说实话,我见过太多人一上来就急着写代码,结果环境没配好,折腾半天连个"Hello World"都跑不出来。做金融大模型更是如此——数据量大、依赖多,环境配不好,后面全是坑。

这一章,我带你把基础环境搭好。跟着我的节奏来,半小时搞定。

Python基础环境搭建

Python版本怎么选?我个人习惯用3.8或3.9。为什么?因为很多金融相关的库(比如TA-Lib、PyPortfolioOpt)在3.10以上版本偶尔会出兼容问题。你想想看,项目做到一半发现某个库装不上,多闹心。

下载地址:python.org

安装时记得勾选"Add Python to PATH"。这个选项我吃过亏——第一次装的时候没勾,后面在命令行里打python死活找不到,折腾了半小时才发现。

注意:Windows用户安装时,路径不要带中文或空格。我曾经有个学员,用户名是"张三",结果Python装到C:\Users\张三\下,后面一堆库报路径错误。

装完后验证一下:

python --version
pip --version

能看到版本号,说明装好了。

Anaconda安装与配置

Anaconda是什么?说白了就是一个Python的"全家桶"。它帮你管理不同项目的环境,避免库版本打架。

我在做量化回测时,一个项目用TensorFlow 1.x,另一个用2.x。如果没有Anaconda,这两个项目根本没法共存。有了conda环境,各玩各的,互不干扰。

下载地址:anaconda.com

安装时注意:

  • 选"Just Me"安装,别选"All Users"
  • 安装路径同样不要有中文
  • 安装过程中会问是否添加PATH,建议勾上

装完后,打开命令行(或Anaconda Prompt),试试:

conda --version

接下来,我建议你创建一个专门用于金融大模型的环境:

conda create -n fin_llm python=3.9
conda activate fin_llm

为什么要单独建环境?嗯,这里要注意——金融大模型依赖的库版本比较特殊,比如transformers、torch这些,版本不对直接报错。单独建个环境,出了问题删了重来,不影响系统Python。

小技巧:我习惯在环境名上加前缀,比如"fin_"表示金融相关,"llm_"表示大模型相关。项目多了你就知道这有多好用了。

Jupyter Notebook使用

Jupyter Notebook是做什么的?它让你能边写代码边看结果,特别适合做数据探索和模型实验。

安装很简单:

pip install jupyter
# 或者用conda
conda install jupyter

启动:

jupyter notebook

浏览器会自动打开一个页面。点"New" -> "Python 3",就能新建一个Notebook了。

我刚开始用Jupyter时,有个坏习惯——把所有代码写在一个单元格里。结果跑起来报错,根本不知道哪行出了问题。后来学乖了:一个单元格只做一件事。比如:

  • 单元格1:导入库
  • 单元格2:加载数据
  • 单元格3:数据清洗
  • 单元格4:模型训练

这样调试起来特别方便,哪一步出问题一目了然。

快捷键必记:
  • Shift + Enter:运行当前单元格并跳到下一个
  • Ctrl + Enter:运行当前单元格但不跳转
  • Esc + A:在当前单元格上方插入新单元格
  • Esc + B:在当前单元格下方插入新单元格
  • Esc + M:将单元格切换为Markdown模式(写说明文字用)

还有一个我踩过的坑——Notebook里的变量是全局共享的。如果你不小心把一个大模型加载到内存里,然后删了某个单元格,模型还在内存里。我曾经因为这个把16G内存跑爆了。解决办法:重启Kernel(Kernel -> Restart)。

GPU环境说明

做金融大模型,GPU几乎是必需品。为什么?因为大模型动辄几亿参数,用CPU跑一个训练可能要几天,GPU几个小时就搞定了。

先看看你的电脑有没有NVIDIA显卡:

nvidia-smi

如果显示显卡信息,说明有NVIDIA GPU。如果没有,也别急——可以用Google Colab或者云GPU服务。

安装CUDA和cuDNN:

  • CUDA:NVIDIA的并行计算平台
  • cuDNN:深度学习加速库

版本匹配很重要。我见过有人装了CUDA 11.8,但PyTorch要求CUDA 11.7,结果死活跑不起来。建议先查PyTorch官网,看它支持哪个CUDA版本,再对应安装。

安装PyTorch(GPU版):

pip install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu118

验证GPU是否可用:

import torch
print(torch.cuda.is_available())  # 返回True说明GPU可用
print(torch.cuda.get_device_name(0))  # 显示显卡型号
避坑指南:我曾经在Windows上装GPU版PyTorch,装完发现torch.cuda.is_available()返回False。查了半天,发现是Visual Studio的C++编译环境没装。解决方案:安装"Microsoft Visual C++ Redistributable"。这个坑我踩了两次才记住。

如果你用的是Mac(M1/M2芯片),也有GPU加速方案:

pip install torch torchvision torchaudio

Mac的MPS加速:

import torch
print(torch.backends.mps.is_available())  # 返回True说明可用

本章知识体系

下面这张图,帮你理清整个环境准备的核心逻辑:

环境准备知识体系 金融大模型开发环境 Python基础环境 Anaconda安装配置 Jupyter Notebook GPU环境说明 版本选择(3.8/3.9) PATH配置 pip验证 环境隔离 conda create 环境激活/删除 安装与启动 单元格操作 快捷键与调试 nvidia-smi检测 CUDA/cuDNN安装 PyTorch GPU验证 环境配好,后面事半功倍 遇到问题先查版本兼容性

环境准备这块,说白了就是"一次配置,长期受益"。我见过太多人因为环境问题浪费一两天时间,最后发现只是某个库版本不对。所以,耐心配好环境,后面你会感谢自己的

我的建议:配好环境后,把安装命令和版本号记下来。下次重装系统或者换电脑,直接复制粘贴,省时省力。我自己就维护了一个"环境配置备忘录",已经用了三年。

专注资料整理