win11 + insightface + pytorch + CUDA + cuDNN 实战安装

部署运行你感兴趣的模型镜像

安装攻关秘籍,步骤如下:
第一步. 下载 pycharm 社区版
官网在这里:https://www.jetbrains.com/pycharm/download/?section=windows

第二步. 下载 anaconda 或者miniconda
// 参考下面文章来执行安装anaconda, 本项目需要python3.10
Windows下Miniconda+Pytorch+Pycharm开发环境搭建指南
anconda或者miniconda官网见下面:
https://docs.conda.io/projects/conda/en/stable/user-guide/install/index.html
https://repo.anaconda.com/archive/
https://repo.anaconda.com/miniconda/

第三步. 配置 anaconda 的path和加速下载源
加速下载源 ,见 Windows下Miniconda+Pytorch+Pycharm开发环境搭建指南

第四步. 按如下步骤来下载依赖包
下面细分几个part. 如果主机有英伟达GPU,则需要执行这个 PART0 的步骤.
PART0 安装GPU依赖
有GPU则必需选,否则就可以跳过!
英伟达官方推荐如下两种方式安装。

方式1: Windows安装 CUDA和cuDNN
1:install CUDA 11.8

for Windows:
install guide : https://developer.nvidia.com/cuda-downloads
测试工具: bandwidthTest.exe, nvcc -V

for Linux:
install guide : https://docs.nvidia.com/cuda/cuda-installation-guide-linux/index.html

download page:
https://developer.nvidia.com/cuda-toolkit-archive
https://developer.nvidia.com/cuda-downloads

2: install cuDNN for Cuda 11.x
install guide https://docs.nvidia.com/deeplearning/cudnn/install-guide/index.html#install-windows

download page: https://developer.nvidia.com/rdp/cudnn-archive
test CMD: deviceQuery.exe

构建测试命令,则到这里去下载源码和构建 https://github.com/nvidia/cuda-samples

在安装了CUDA toolkit后,测试工具 bandwidthTest 和 deviceQuery
就在下面的目录下:C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\vxx\extras\demo_suite\

方式2:用conda安装 CUDA和cuDNN
conda config --add channels conda-forge
conda config --set channel_priority strict
conda install cudatoolkit11.8 -c anaconda
conda install cudnn
8.4.1.50 -c anaconda --force //8.5.0.96 在conda上面没有。它有cudnn 8.9.2.26 cuda11_0

CUDA 和cuDNN 参考文档

英伟达安装cuda

 安装文档: https://docs.nvidia.com/cuda/cuda-installation-guide-microsoft-windows/index.html#install-cuda-software ,  
  CUDA 下载 https://developer.nvidia.com/cuda-11-8-0-download-archive

英伟达安 cuDNN

   安装文档: https://docs.nvidia.com/deeplearning/cudnn/install-guide/index.html
  cuDNN 下载 https://developer.nvidia.com/rdp/cudnn-archive

网络文章

https://cloud.tencent.com/developer/article/2153193
https://developer.nvidia.com/cuda-11-8-0-download-archive

PART1 安装Microsoft Visual C++ 14.0 或者以上
参考 https://visualstudio.microsoft.com/visual-cpp-build-tools/
点击:下载生成工具,下载 vs_BuildTools.exe到本地,然后按屏幕提示一步步走就可以了,不需要修改
这一步不装,则会在pip install insightface==0.7.3碰到错误。
在安装过程中,确保选择了“C++ 编译工具”选项,以及任何必要的 Windows SDK 和其他依赖组件。
完成安装后,重新启动计算机以确保所有路径和环境变量更新生效

PART2 安装别的依赖包

#强烈建议在anaconda中,用它的pip来安装而不用conda来安装
#进入conda的命令行窗口,执行下面的命令
pip config set global.index-url https://pypi.tuna.tsinghua.edu.cn/simple
pip config list

conda create -n ai_py310 python3.10
conda activate ai_py310
pip config set global.index-url https://pypi.tuna.tsinghua.edu.cn/simple
pip config list
#pytorch会安装numpy,cudnn
conda install pytorch
2.0.1 torchvision0.15.2 torchaudio2.0.2 pytorch-cuda=11.8 -c pytorch -c nvidia
pip install onnxruntime-gpu1.15.1
pip install tqdm
pip install redis
pip install boto3
pip install opencv-python
pip install python-multipart
pip install fastapi
pip install uvicorn
pip install pillow
pip install loguru
pip install psutil
FAQ:
#如果有问题,则要装依赖。
#pip install blosc2
2.0.0
#碰到过下面的问题,
//error: Microsoft Visual C++ 14.0 or greater is required. Get it with “Microsoft C++ Build Tools”: https://visualstudio.microsoft.com/visual-cpp-build-tools/ [end of output]
//则安装vs studio professional,下载 vs_BuildTools.exe,然后执行即可,同时要注意下面的步骤
在安装过程中,确保选择了“C++ 编译工具”选项,以及任何必要的 Windows SDK 和其他依赖组件。
完成安装后,重新启动计算机以确保所有路径和环境变量更新生效

英伟达的cudatoolkit 和 cudnn 不能用pip来安装,可以选择用windows或者conda来安装

用conda安装 英伟达的cudatoolkit 和 cudnn (可选)

PART3 安装ffmpeg工具
如果涉及到视频处理,则需要安装fmpeg工具,去官网上下载windows工具即可。
windows环境下:将ffmpeg-6.0-full_build.7z解压,将bin下的ffmpeg.exe和ffprobe.exe 添加至环境变量。将然后将这两个复制到你pytorch环境下【和torch下的python在一个目录中即可】
https://ffmpeg.org//download.html
第五步. 配置pycharm, 下载项目工程代码
配置conda为pycharm的python解释器 (新加解释器),下载自己的工程代码,运行和调试代码即可。
windows环境下 ffmpeg.exe和ffprobe.exe 添加至环境变量。将然后将这两个复制到你pytorch环境下
linux环境下 可以直接安装即可,如用 apt-get install ffmpeg, yum install ffmpeg等
本工程用的 ffmpeg-6.0-full_build
第六步. 结束

参考资料

cudatoolkit 和 cudnn 不能用pip来安装

  1. pytorch和cuda的版本映射 https://pytorch.org/get-started/previous-versions/ // 用CUDA 11.8
  2. onnx,CUDA和cuDNN的关系 https://onnxruntime.ai/docs/execution-providers/CUDA-ExecutionProvider.html
    #8.5.0.96 (Windows), Tested with CUDA versions from 11.6 up to 11.8, and cuDNN from 8.2.4 up to 8.7.0
    3.英伟达安装cuda 文档
    https://docs.nvidia.com/cuda/cuda-installation-guide-microsoft-windows/index.html#install-cuda-software
    https://developer.nvidia.com/cuda-11-8-0-download-archive
  3. 好的网络文章
    https://cloud.tencent.com/developer/article/2153193
    insightface 官网项目 https://insightface.ai/
    ffmpeg 官网 https://ffmpeg.org//download.html
    conda 官网 https://docs.conda.io/projects/conda/en/stable/index.html
  4. anaconda 官网 https://anaconda.cloud/, 历史包 https://repo.anaconda.com/archive/

您可能感兴趣的与本文相关的镜像

PyTorch 2.5

PyTorch 2.5

PyTorch
Cuda

PyTorch 是一个开源的 Python 机器学习库,基于 Torch 库,底层由 C++ 实现,应用于人工智能领域,如计算机视觉和自然语言处理

### 系统环境准备 在安装 CUDA 12.9、cuDNNPythonPyTorch 之前,确保系统满足以下要求: - **操作系统**:Windows 10 或 Windows 11(64 位) - **NVIDIA 显卡驱动**:需要安装CUDA 12.9 兼容的 NVIDIA 显卡驱动,建议使用最新版本的驱动程序[^3]。 - **Python 环境**:建议使用 Python 3.10 或更高版本,并使用虚拟环境管理工具如 `conda` 或 `venv` 来隔离项目依赖。 --- ### 安装 CUDA Toolkit 12.9 1. **检查显卡支持的 CUDA 版本** 在命令行中输入以下命令以查看显卡支持的最高 CUDA 版本: ```bash nvidia-smi ``` 输出中的 `CUDA Version` 字段将显示当前显卡支持的最高 CUDA 版本。如果支持 CUDA 12.9,则继续安装;否则,需要升级显卡驱动[^3]。 2. **下载并安装 CUDA Toolkit 12.9** 前往 [CUDA Toolkit 官方下载页面](https://developer.nvidia.com/cuda-toolkit-archive),选择对应操作系统的 CUDA 12.9 版本进行安装[^4]。 3. **验证 CUDA 安装** 安装完成后,打开命令行并执行以下命令以验证 CUDA 是否安装成功: ```bash nvcc --version ``` 输出应显示 CUDA 编译器驱动程序版本为 12.9。 --- ### 安装 cuDNN 9.9.0 1. **下载 cuDNN** 前往 [NVIDIA cuDNN 官方页面](https://developer.nvidia.com/cudnn),登录后下载与 CUDA 12.9 兼容的 cuDNN 版本(推荐 cuDNN 9.9.0)。 2. **解压并复制文件** 将下载的 cuDNN 压缩包解压后,将 `bin`、`include` 和 `lib` 文件夹中的内容分别复制到 CUDA 安装目录下的对应文件夹中。例如,将 `x64` 文件夹中的内容复制到 `C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v12.9\lib\x64` 中[^5]。 3. **验证 cuDNN 安装** 由于 cuDNN 没有独立的命令行工具进行验证,可以通过后续安装PyTorch 测试其是否正常工作。 --- ### 安装 PythonPyTorch 1. **安装 Python** 前往 [Python 官方网站](https://www.python.org/downloads/) 下载并安装最新版本的 Python(建议使用 Python 3.10 或更高版本)。安装过程中请勾选 "Add to PATH" 选项。 2. **创建虚拟环境** 使用 `venv` 创建一个虚拟环境: ```bash python -m venv pytorch_env pytorch_env\Scripts\activate ``` 3. **安装 PyTorch** 目前,PyTorch 官方尚未发布支持 CUDA 12.9 的版本。因此,建议使用 `conda` 安装 PyTorch,它会自动处理与 CUDA 的兼容性问题: ```bash conda install pytorch torchvision torchaudio cudatoolkit=12.8 -c pytorch ``` 如果希望尝试使用 `pip` 安装 PyTorch,可以访问 [PyTorch 官方安装页面](https://pytorch.org/get-started/locally/) 获取适配 CUDA 12.8 的安装命令。由于 CUDA 12.9 尚未被官方支持,不建议直接安装最新版本的 PyTorch[^2]。 4. **验证 PyTorch 安装** 打开 Python 解释器并运行以下代码以验证 PyTorch 是否成功安装并能够使用 CUDA: ```python import torch print(torch.__version__) print(torch.cuda.is_available()) print(torch.backends.cudnn.enabled) ``` 如果输出显示 `True`,则表示 PyTorch 已成功安装并能够使用 CUDAcuDNN。 --- ### 配置总结 - **CUDA Toolkit 12.9** 已成功安装,并且可以通过 `nvcc --version` 进行验证。 - **cuDNN 9.9.0** 已解压并复制到 CUDA 安装目录中,确保 PyTorch 能够正常使用。 - **Python** 和 **PyTorch** 已通过虚拟环境安装,并且可以通过简单的 Python 脚本验证其是否能够正确使用 CUDAcuDNN。 --- ###
评论
成就一亿技术人!
拼手气红包6.0元
还能输入1000个字符
 
红包 添加红包
表情包 插入表情
 条评论被折叠 查看
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值