AI环境搭建--windows

安装conda

• 打开下载地址: https://www.anaconda.com/download，选择skip registration
  

• 下载安装:有2个版本Anaconda Installers和Miniconda Installers，可以选择mini版本。
  

安装完成后,打开命令行cmd

#输入命令
conda -V

#返回结果
conda 24.11.3

安装CUDA

CUDA是为了微调模型参数时，使用NVIDIA卡的加速计算，若没有N卡，不需要安装CUDA
查看NVIDIA卡的支持信息

#执行命令
nvidia-smi

• 选择环境
  支持12.6及以下的CUDA。
  因为要安装pytorch，需要从pytorch查看支持哪个版本，再安装。

CUDA地址：https://developer.nvidia.com/cuda-toolkit

#执行命令
nvcc -V

#返回信息
nvcc: NVIDIA (R) Cuda compiler driver
Copyright (c) 2005-2024 NVIDIA Corporation
Built on Thu_Mar_28_02:30:10_Pacific_Daylight_Time_2024
Cuda compilation tools, release 12.4, V12.4.131
Build cuda_12.4.r12.4/compiler.34097967_0

• 安装cudnn
  下载地址：https://developer.nvidia.com/cudnn
  注：需要先注册再下载，安装时选择与CUDA相对就的版本
  下载解压后，会有3个文件夹bin,include,lib。把这3个文件夹复制到CUDA的文件下。

• CUDA ZIP包下载
  选择版本：https://developer.nvidia.com/cuda-toolkit-archive
  

打开地址：https://developer.download.nvidia.cn/compute/cuda/redist/
查看对就版本的发布文件：https://developer.download.nvidia.cn/compute/cuda/redist/redistrib_12.4.1.json

根据平台下载所有相关的zip包，解压到固定位置 ,放在了D:\CUDA

• CUDA环境变量

#增加环境变量CUDA_PATH
CUDA_PATH=D:\CUDA

#PATH中添加
D:\CUDA\bin
D:\CUDA\lib
D:\CUDA\include
D:\CUDA\libnvvp

安装pytorch

• 官网地址：https://pytorch.org/
  

根据选择的选项会生成安装命令，执行命令安装pytorch

pip3 install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu124

确认安装

#执行命令
pip list |find "torch"

#返回信息
torch                             2.6.0+cu124
torchaudio                        2.6.0+cu124
torchvision                       0.21.0+cu124

开发环境pycharm

• 官网地址：https://www.jetbrains.com/pycharm/download/
  可以安装社区版