VideoAgentTrek Screen Filter 环境配置详解:Ubuntu系统下的Anaconda与CUDA部署
VideoAgentTrek Screen Filter 环境配置详解Ubuntu系统下的Anaconda与CUDA部署如果你刚接触AI视频处理想在Ubuntu系统上跑起VideoAgentTrek Screen Filter这样的工具第一步也是最关键的一步就是把环境搭好。我见过不少朋友代码和模型都准备好了结果卡在环境配置上一折腾就是好几天。今天这篇我就带你一步步在Ubuntu 20.04 LTS上把VideoAgentTrek Screen Filter需要的AI开发环境给配齐了。核心就四件事用Anaconda管好Python环境装上正确的NVIDIA驱动和CUDA安装PyTorch这些深度学习框架最后验证GPU能不能用起来。我会把过程中容易踩的坑和解决办法都告诉你让你一次搞定少走弯路。1. 准备工作与系统检查在开始安装之前我们先花几分钟把“地基”打好。这就像盖房子前先看看图纸和材料能避免后面很多麻烦。首先确认一下你的Ubuntu版本。虽然我们以20.04 LTS为例但其他版本如18.04, 22.04的步骤也大同小异。打开终端输入下面这个命令lsb_release -a你会看到类似Description: Ubuntu 20.04.6 LTS的输出这就对了。LTS版本长期支持比较稳定推荐使用。接下来更新一下系统软件包列表确保我们安装的都是最新可用的版本。这步很重要能避免一些依赖冲突。sudo apt update然后安装一些后续步骤可能会用到的工具比如下载用的wget和压缩包处理工具。sudo apt install wget -y最后检查一下你的显卡。VideoAgentTrek Screen Filter这类视频AI工具通常需要GPU来加速所以你得有一块NVIDIA的显卡。输入lspci | grep -i nvidia如果能看到你的显卡型号信息比如NVIDIA Corporation GA102 [GeForce RTX 3080]那就没问题。如果什么都没显示那可能你的机器没有NVIDIA显卡或者需要先安装基础的显卡驱动。对于没有独立GPU的机器虽然也能用CPU运行但速度会慢很多体验大打折扣。2. 安装与配置AnacondaPython环境管理是AI开发里的头等大事。不同的项目可能需要不同版本的Python甚至不同版本的库混在一起很容易出问题。Anaconda就是一个帮你把不同环境隔离开的工具非常省心。2.1 下载Anaconda安装脚本我们去Anaconda官网找适合Linux的安装脚本。通常选择最新的Python 3.x版本就好。在终端里用wget命令直接下载。wget https://repo.anaconda.com/archive/Anaconda3-2024.02-1-Linux-x86_64.sh注意上面的链接版本号2024.02-1可能会随着时间更新。你可以去 Anaconda官网 查看最新的Linux安装包链接替换掉命令里的地址。下载完成后先验证一下文件的完整性这是个好习惯。sha256sum Anaconda3-2024.02-1-Linux-x86_64.sh把终端里显示的一长串哈希值和官网这个文件旁边提供的SHA256校验码对比一下一致就可以放心安装了。2.2 运行安装程序运行下面的命令开始安装bash Anaconda3-2024.02-1-Linux-x86_64.sh安装过程中你会看到许可协议一直按回车翻到最后输入yes表示同意。 接着会问你安装路径直接回车就安装在默认的~/anaconda3目录下。 最后安装程序会问你是否要初始化Anaconda3这里一定要输入yes。这步会让Anaconda修改你的shell配置文件比如~/.bashrc把conda命令加到系统路径里。安装完成后关掉当前终端再重新打开一个新的终端窗口。这是为了让刚才的配置生效。然后你可以用下面的命令验证是否安装成功conda --version如果显示出版本号例如conda 24.1.2恭喜你Anaconda安装成功了。2.3 为项目创建独立的Python环境现在我们来创建一个专门给VideoAgentTrek Screen Filter使用的环境这样它需要的库不会影响你系统里其他项目。conda create -n videoagent python3.9 -y这里-n videoagent指定了环境名叫videoagentpython3.9指定安装Python 3.9版本。Python 3.8或3.10通常也可以但3.9是目前很多AI框架兼容性比较好的一个版本。创建好后激活这个环境conda activate videoagent激活后你会发现终端提示符前面变成了(videoagent)这表示你现在已经在这个独立的环境里操作了。之后所有包的安装都只在这个环境内生效。3. 安装NVIDIA驱动与CUDA工具包这是让GPU能参与AI计算的核心步骤。驱动是让系统认识你的显卡CUDA则是让PyTorch这些框架能调用显卡干活的工具包。3.1 安装NVIDIA显卡驱动Ubuntu系统自带的“软件和更新”工具里可以很方便地安装驱动。在系统菜单里找到它打开后切换到“附加驱动”标签页。 你会看到一个列表里面是系统检测到的可用的NVIDIA驱动版本。通常选择后面标有“专有”或“proprietary”以及“推荐”的版本例如nvidia-driver-535然后点击“应用更改”。系统会自动下载并安装。或者你也可以在终端里用命令安装比如安装535版本的驱动sudo apt install nvidia-driver-535 -y安装完成后必须重启电脑。sudo reboot重启后再次打开终端输入以下命令验证驱动是否安装成功nvidia-smi这个命令会输出一个表格显示你的GPU型号、驱动版本、CUDA版本这里显示的是驱动支持的最高CUDA版本并非已安装的以及GPU的使用情况。看到这个表格就说明驱动装好了。3.2 通过Conda安装CUDA工具包接下来安装CUDA。对于PyTorch用户我强烈推荐通过Conda来安装CUDA而不是去NVIDIA官网下载那个巨大的.run安装包。Conda方式安装的CUDA是独立于系统环境的不会影响系统其他部分管理和卸载也方便。首先确保你已经激活了之前创建的videoagent环境。然后使用conda命令安装CUDA 11.8这是目前与PyTorch稳定版兼容性很好的一个版本conda install cudatoolkit11.8 -c conda-forge -y-c conda-forge指定从conda-forge这个频道下载软件更全更新。安装需要一些时间耐心等待。安装完成后可以验证一下nvcc --version如果显示出版本信息如release 11.8并且路径是在你的Anaconda环境里例如/home/你的用户名/anaconda3/envs/videoagent/bin/nvcc那就说明CUDA工具包已经正确安装到当前环境了。4. 安装PyTorch及其他Python依赖环境搭好了现在该把主角——深度学习框架请进来了。VideoAgentTrek Screen Filter这类项目通常基于PyTorch。4.1 安装PyTorch去 PyTorch官网 看看选择适合你的配置。因为我们用Conda安装了CUDA 11.8所以对应选择CUDA 11.8的安装命令。在激活的videoagent环境下运行pip install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu118这里使用pip安装并指定了针对CUDA 11.8的索引地址。安装完成后我们来写个简单的Python脚本验证PyTorch能否识别GPU。创建一个叫test_gpu.py的文件内容如下import torch print(fPyTorch版本: {torch.__version__}) print(fCUDA是否可用: {torch.cuda.is_available()}) if torch.cuda.is_available(): print(fGPU设备名称: {torch.cuda.get_device_name(0)}) print(f当前CUDA版本: {torch.version.cuda}) else: print(未检测到可用GPU请检查CUDA和驱动安装。)然后在终端运行它python test_gpu.py你期待的完美输出应该是CUDA是否可用: True显示你的GPU型号如NVIDIA GeForce RTX 3080显示CUDA版本如11.8如果显示为False别慌回头检查驱动是否安装并重启以及CUDA工具包是否安装在了当前conda环境里。4.2 安装其他必要依赖VideoAgentTrek Screen Filter可能还需要一些常见的视频处理和工具库。我们可以先安装一批pip install opencv-python pillow numpy scipy tqdmopencv-python: 处理视频和图像的核心库。pillow: Python的图像处理库。numpy: 科学计算基础库几乎所有AI框架都依赖它。scipy: 用于高级数学计算。tqdm: 在终端显示进度条让你知道程序还在跑。具体到VideoAgentTrek Screen Filter它很可能还有自己特定的依赖项。这些通常会在项目的requirements.txt文件或README.md中列出。假设你拿到了这个文件在项目目录下运行下面这个命令就能一键安装所有依赖pip install -r requirements.txt5. 常见问题与故障排除环境配置很少有一帆风顺的这里我总结几个最常见的问题和解决办法。问题一conda: command not found这说明Anaconda没有正确初始化。重新打开终端执行source ~/.bashrc或者直接关掉终端再开一个新的。如果还不行检查你的~/.bashrc文件末尾是否有Anaconda添加的类似这样的行# conda initialize ... # 这里有一些conda init添加的内容 # conda initialize 如果没有你可能需要手动运行~/anaconda3/bin/conda init bash然后重启终端。问题二nvidia-smi命令找不到或没输出首先确认电脑已经重启。如果重启后还不行可能是驱动没装上。回到“软件和更新”的“附加驱动”里换个驱动版本试试比如从535换成470应用更改并重启。也可以尝试用终端命令彻底删除再重装sudo apt purge nvidia* -y sudo apt autoremove -y sudo apt install nvidia-driver-535 -y sudo reboot问题三PyTorch显示CUDA不可用但nvidia-smi正常这通常是PyTorch版本和CUDA版本不匹配或者PyTorch安装的不是GPU版本。请确保在正确的conda环境videoagent里安装PyTorch。使用的PyTorch安装命令明确指定了CUDA版本如我们用的cu118。可以尝试先卸载PyTorch再重装pip uninstall torch torchvision torchaudio然后重新执行官网提供的正确安装命令。问题四环境冲突或包版本不兼容这就是为什么我们要用conda创建独立环境的原因。如果在这个环境里搞乱了最简单的办法就是删掉重建conda deactivate # 先退出当前环境 conda remove -n videoagent --all -y # 删除整个环境 conda create -n videoagent python3.9 -y # 重新创建 conda activate videoagent # 然后重新安装CUDA和PyTorch6. 总结与后续步骤走完上面这些步骤你的Ubuntu系统上应该已经有一个为AI视频处理准备好的独立Python环境了。Anaconda帮你管着环境NVIDIA驱动让系统认出了显卡CUDA给了PyTorch调用显卡的能力最后PyTorch自己也成功装上了。你可以再运行一次那个test_gpu.py脚本看到一切正常的输出心里就踏实了。环境配好只是第一步就像是把厨房和厨具都备齐了。接下来你就可以去获取VideoAgentTrek Screen Filter的模型代码按照它的项目说明在这个我们刚搭好的videoagent环境里运行它了。通常步骤是克隆代码仓库然后安装项目特定的依赖最后尝试运行示例。如果项目有提供示例视频或脚本先用小的例子跑通再处理你自己的视频材料。配置环境是个需要点耐心的活儿遇到报错多查查错误信息大部分问题网上都有解决方案。最重要的是保持环境隔离一步一步来验证。希望这篇详细的指南能帮你顺利跨过这第一步。获取更多AI镜像想探索更多AI镜像和应用场景访问 CSDN星图镜像广场提供丰富的预置镜像覆盖大模型推理、图像生成、视频生成、模型微调等多个领域支持一键部署。