目的

本文搭建一个AI模型训练环境，用于AI模型训练。系统win10专业版，显卡为索泰4060TI 16G。

过程

1、安装python

版本：python 3.12.3

2、安装ffmpeg

安装方式一：

下载列表：https://github.com/BtbN/FFmpeg-Builds/releases 官网下载编译好的二进制文件。链接地址：ffmpeg 下载完成后解压，在环境变量path中添加解压的ffmpeg目录下bin目录路径。

安装方式二：

windows平台下可以通过命令行下载ffmpeg；可以使用MSYS2/MSYS2 MINGW64来下载，如下所示。

pacman -S mingw-w64-x86_64-ffmpeg
如果需要libx264、libx265、libfdk-aac则需要下载相应的库，下载命令如下：

pacman -S mingw-w64-x86_64-fdk-aac //下载libfdk-aac
pacman -S mingw-w64-x86_64-x264 //下载libx264
pacman -S mingw-w64-x86_64-x265 //下载libx265
ffmpeg下载安装完成后可以通过ffmpeg.exe -version可以查看编译后的ffmpeg信息

3、安装cuda

a、查看显卡支持的cuda版本，在cmd命令行下执行nvidia-smi 这里显示最大支持12.4版本cuda。

b、下载最新版本cuda

下载链接：cuda下载页面 cuda12.4版本

c、下载完成后直接安装即可，安装目录默认。

安装类型选择自定义，第一次安装下一步勾选全部可选择的工具，一直默认下一步到安装结束

d、查看环境变量是否设置成功

path名称的环境变量都是指向cuda安装目录下。 下方两个v10.0版本是由于开始装了一个v10版本cuda，这里暂时没有发现有影响。

4、安装cuDNN

下载地址：https://developer.nvidia.com/cudnn-downloads a、确认要下载的cuDNN；官网有说明跟cuda的匹配版本。

我们下载的cudaV12.4，所以cuDNN下载9.1.1版本

下载完成后是exe执行文件，直接安装默认配置选项即可。

5、验证环境是否成功，分两步验证。

a、找到安装目录文件CUDA\v12.4\extras\demo_suite>bandwidthTest.exe，cmd命令执行exe文件：如下显示则成功

b、找到安装目录文件 CUDA\v12.4\extras\demo_suite>deviceQuery.exe，cmd命令执行exe文件：如下显示则成功

6、安装pytorch

Pytorch版本与cuda版本必须要对应，不然无法应用 下载地址：Pytorch

验证：

在进入python环境输入如下命令返回相关内容则安装成功

import torch
 
print(torch.__version__)
print(torch.cuda.is_available())
print(torch.version.cuda)

部分参考来源：https://zhuanlan.zhihu.com/p/684095163