Anaconda+Pytorch(CUDA)开发环境全配置

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

0000 - 图例 💻

  1. 本文中所有的 🌐 图标指最好用魔法访问该链接
  2. 本文中所有的 🔎 图标指建议使用搜索引擎进一步探究,这里不多赘述。

0001 - 基础环境配置

  1. 安装Anaconda🌐(推荐把安装盘改到D盘,且CD盘均留有至少20G空间)

  2. 配置Anaconda环境变量(在PATH 🔎 中加入以下路径)

    D:\ProgramData\anaconda3
    D:\ProgramData\anaconda3\Scripts
    D:\ProgramData\anaconda3\Library\bin
    D:\ProgramData\anaconda3\Library\mingw-w64\bin
    
  3. 运行以下命令换源(决定用pip后就不要用conda install)

    pip config set global.index-url https://pypi.tuna.tsinghua.edu.cn/simple
    

    除去清华源以外还可以用很多源:

    清华大学 https://pypi.tuna.tsinghua.edu.cn/simple/
    阿里云 http://mirrors.aliyun.com/pypi/simple/
    中国科技大学 https://pypi.mirrors.ustc.edu.cn/simple/
    豆瓣(douban) http://pypi.douban.com/simple/
    中国科学技术大学 http://pypi.mirrors.ustc.edu.cn/simple/
    

    当然,你也可以选择用conda,那之后就不要用pip install(不建议,因为现有教程中用conda的少,有些包版本可能没有)

    conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main/
    conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/free/
    conda config --set show_channel_urls yes
    conda config --show
    

    还可以替换更多部分(一般不选,如果报错就去掉这些源)

    http://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/
    http://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/conda-forge/
    http://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/msys2/
    http://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/bioconda/
    

    除了清华源外还可以选择其他源

    # 阿里源 (最近不能访问)
    https://mirrors.aliyun.com/anaconda/pkgs/main/
    https://mirrors.aliyun.com/anaconda/pkgs/free/
    https://mirrors.aliyun.com/anaconda/cloud/
    https://mirrors.aliyun.com/anaconda/cloud/conda-forge/
    # 中科大源 (部分不能访问)
    https://mirrors.ustc.edu.cn/anaconda/pkgs/main/
    https://mirrors.ustc.edu.cn/anaconda/pkgs/free/
    https://mirrors.ustc.edu.cn/anaconda/cloud/
    https://mirrors.ustc.edu.cn/anaconda/cloud/conda-forge/
    

    如果channel不可用可以用以下命令删除

    conda config --remove channels https://mirrors.aliyun.com/anaconda/pkgs/main/
    conda config --remove channels https://mirrors.aliyun.com/anaconda/pkgs/free/
    
  4. 创建一个虚拟环境(如果报错大概率是网络波动,可以多试几次)

  5. 使用以下命令测试是否安装成功(只支持cmd,不支持Powersshell和Terminal)

    conda activate 环境名
    

0010 - 一般环境配置

  1. 从以前的环境中导出装过的第三方库(若没有可跳过)

    pip freeze > requirements.txt
    

    记得打开requirements.txt== 替换为 >=

  2. 安装Python库大全(用于开发的环境)

    pip install -r requirements.txt --ignore-installed
    
  3. 如果出现冲突就把PACKAGE >= XX.XX.XX替换为PACKAGE == YY.YY.YY(YY.YY.YY为该库的最新稳定版本)如果不强行指定,pip不知为何会默认装次新的版本,从而引发各种冲突。

0011 - NVIDIA显卡环境配置

  1. 访问Pytorch官网查询支持的CUDA版本和安装命令

  2. 访问CUDA下载地址🌐下载正确版本的CUDA(以下几步下载较慢,最好用IDM一类的下载器)

  3. 访问cuDNN下载地址🌐下载CUDA版本对应的cuDNN(可能需要注册为开发者)

  4. 安装CUDA,检查环境变量是否已经写入(默认会自动写入)

  5. 将cuDNN解压并用其中的文件夹覆盖替换CUDA安装目录下的文件夹

  6. 使以下命令确认是否安装成功

    nvCC -V
    

0100 - Pytorch环境配置

  1. 新开一个环境用于安装Torch(下载过程较慢但切不可开代理) 如果中途网络卡顿则会报错,可以在原指令的基础上加一个 -f https://download.pytorch.org/whl/torch_stable.html以提升其稳定性。至于为什么能增加稳定性,我也不知道,就是玄学

    pip3 install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu121 -f https://download.pytorch.org/whl/torch_stable.html
    

    慢慢等吧,祝你好运!(如果不想等的话可以看后面的搭建Torch本地源)

0101 - 安装VScode

  1. 一定要在装好Anaconda之后再进行这一步,否则后面在VScode里面切换conda环境会比较麻烦

  2. 安装VScode,能勾选的全勾选

  3. 进入VScode后先下载中文扩展,在左侧第四个图标点开,搜索Chinese,第一个点击Install

  4. 之后安装其他的扩展,这里罗列以下推荐扩展(语法解析类扩展在打开源码的时候会自动在右下角建议你安装)

    CodeSnap-plus  # 代码截图
    Live Share  # 联机写代码(需登陆GitHub)
    Markdown All in One  # 写Markdown用
    Pylance  # Python必备
    Python  # Python必备
    TONGYI Lingma  # AI代码工具(需登陆阿里云)
    Auto Close Tag  # 自动补全HTML标签
    
  5. 打开一个Python文件,点击界面上方Banner打开信任工作区设置,滚到底,将常用目录添加入信任的工作区

  6. 检查能否在Anaconda的Home页面看到VScode,如果可以就成功了

0110 - 安装Git

  1. 下载git🌐安装包并安装(下载偏慢,建议用IDM)

  2. 鼠标右键 --> Git Bash Here 并设置用户名和邮箱

    git config --global user.name sfkgroup
    git config --global user.email sfkgroup@yeah.net
    

    附上常见的git国内源

    https://kgithub.com  # 相应时间过长
    https://hub.nuaa.cf  # 只支持git
    https://hub.yzuu.cf  # 只支持git
    https://hub.fgit.ml  # 相应时间过长
    https://gitclone.com  # 不支持git
    https://githubfast.com  # 只支持git
    https://github.ur1.fun/  # 只支持DownloadZIP
    

    网站难连,魔法难用,总之加油。git使用参考文献🔎

0111 - 安装GitHub Desktop(可选)

  1. 下载🌐并安装GitHub Desktop(不稳定,用IDM或魔法)
  2. 下载汉化组件🌐(release版本,注意版本是否对应)
  3. 进入目录/app-*.*.*/resources/app/(*是版本号,不一定相同)并替换文件
  4. 打开软件并登陆Github账号(当然,你应该早就注册过对吧🙂)
  5. 日常使用指南 (1🔎) (2🔎)

1000 - 搭建Torch本地源(可选)

  1. 以搭建download.pytorch.org/whl/cu121/源为例

  2. 先看安装torch需要那些.whl(当然,你可以全下载,但是太大了)

    https://download.pytorch.org/whl/cu121/torch-2.3.1%2Bcu121-cp311-cp311-win_amd64.whl
    https://download.pytorch.org/whl/cu121/torchvision-0.18.1%2Bcu121-cp311-cp311-win_amd64.whl
    https://download.pytorch.org/whl/cu121/torchaudio-2.3.1%2Bcu121-cp311-cp311-win_amd64.whl
    https://download.pytorch.org/whl/filelock-3.13.1-py3-none-any.whl
    https://download.pytorch.org/whl/typing_extensions-4.9.0-py3-none-any.whl
    https://download.pytorch.org/whl/sympy-1.12-py3-none-any.whl
    https://download.pytorch.org/whl/networkx-3.2.1-py3-none-any.whl
    https://download.pytorch.org/whl/Jinja2-3.1.3-py3-none-any.whl
    https://download.pytorch.org/whl/fsspec-2024.2.0-py3-none-any.whl
    https://download.pytorch.org/whl/mkl-2021.4.0-py2.py3-none-win_amd64.whl
    https://download.pytorch.org/whl/pillow-10.2.0-cp311-cp311-win_amd64.whl
    https://download.pytorch.org/whl/intel_openmp-2021.4.0-py2.py3-none-win_amd64.whl
    https://download.pytorch.org/whl/tbb-2021.11.0-py3-none-win_amd64.whl
    https://download.pytorch.org/whl/MarkupSafe-2.1.5-cp311-cp311-win_amd64.whl
    https://download.pytorch.org/whl/mpmath-1.3.0-py3-none-any.whl
    https://download.pytorch.org/whl/numpy-1.26.3-cp311-cp311-win_amd64.whl
    
  3. 安装pip2pi(如果是3.7以下的Python,装0.7.0及以下的pip2pi,反之只能装0.8.0及以上版本的)

    pip install pip2pi==0.8.2
    

    如果之后运行中出现此类报错,那么一定是你的版本没有与Python版本对应

    AttributeError: module 'cgi' has no attribute 'escape'
    
  4. 用下载的文件构建源文件目录

    │  mk_index.py
    │  http-server.exe
    │
    ├─pypi-packages
    │  │  filelock-3.13.1-py3-none-any.whl
    │  │  fsspec-2024.2.0-py3-none-any.whl
    │  │  intel_openmp-2021.4.0-py2.py3-none-win_amd64.whl
    │  │  Jinja2-3.1.3-py3-none-any.whl
    │  │  MarkupSafe-2.1.5-cp311-cp311-win_amd64.whl
    │  │  mkl-2021.4.0-py2.py3-none-win_amd64.whl
    │  │  mpmath-1.3.0-py3-none-any.whl
    │  │  networkx-3.2.1-py3-none-any.whl
    │  │  pillow-10.2.0-cp311-cp311-win_amd64.whl
    │  │  sympy-1.12-py3-none-any.whl
    │  │  tbb-2021.11.0-py3-none-win_amd64.whl
    │  │  typing_extensions-4.9.0-py3-none-any.whl
    |  |  numpy-1.26.3-cp311-cp311-win_amd64.whl
    │
    └─torch-packages
        │  torch-2.3.1+cu121-cp311-cp311-win_amd64.whl
        │  torchaudio-2.3.1+cu121-cp311-cp311-win_amd64.whl
        │  torchvision-0.18.1+cu121-cp311-cp311-win_amd64.whl
    

    http-server.exe为服务器目录,mk_index.pyWindows下批量创建HTML索引的程序(如下)

    import os 
    import glob
    import shutil
    
    if os.path.exists("./torch-packages/simple"):shutil.rmtree("./torch-packages/simple")
    if os.path.exists("./pypi-packages/simple"):shutil.rmtree("./pypi-packages/simple")
    for file in os.listdir("./torch-packages"):
        name = file.split("-")[0]
        if not os.path.exists(f"./torch-packages/{name}"):
            os.mkdir(f"./torch-packages/{name}")
            shutil.copy(f"./torch-packages/{file}",f"./torch-packages/{name}/{file}")
    os.system(f'dir2pi ./torch-packages')
    os.system(f'dir2pi ./pypi-packages')
    html_files = glob.glob("./pypi-packages/simple/*/index.html")
    for html_path in html_files:
        with open(html_path,"r") as f:
            lines=f.readlines() 
            data=[]  
            for i in lines:
                if("<a href='" in i):
                    i=i.replace("<a href='","<a href='../../")
                data.append(i)
        with open(html_path,"w") as f:
            for i in data:
                f.writelines(i)
        print(html_path)
    
  5. 安装Pytorch

    pip3 install torch torchvision torchaudio --index-url http://127.0.0.1:8080/torch-packages -f http://127.0.0.1:8080/pypi-packages --trusted-host 127.0.0.1
    

1001 - StableDiffusion配置

  1. 下载SD的Release版并解压

  2. 用VScode打开文件夹并将*.py,*.bat,*.sh中的github.com全部替换为githubfast.com, huggingface.co替换为hf-mirror.com

  3. 搜索torch==找到torch安装指令,可以将后面替换成本地源

  4. 下载huggingface上的openai/clip-vit-large-patch14并保存至.\openai/clip-vit-large-patch14

  5. 下载一些模型并放入.\models\Stable-diffusion目录

  6. 开始安装StableDiffusion

    webui.bat --skip-torch-cuda-test --lowvram --precision full --no-half  --disable-safe-unpickle --disable-nan-check
    

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

PyTorch 2.5

PyTorch 2.5

PyTorch
Cuda

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

评论 1
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值