注:本人使用的是 RTX2060 - 6G版
特别提醒:安装一定要 CUDA 和 PyTorch 版本能配套用,不然会有生成保存问题(我是这样的),装完用 python -m xformers.info 这个看对应的版本
建议:有些命令安装在venv 虚拟机中做,这样东西都在venv对应目录
在最后有些资源,我在对应官网下了放在百度网盘,有些下的慢的,可以去里面下
安装Git
https://registry.npmmirror.com/-/binary/git-for-windows/v2.38.1.windows.1/Git-2.38.1-64-bit.exe
安装python3.10.6
官方的
https://www.python.org/ftp/python/3.10.6/python-3.10.6-amd64.exe
华为代理的
https://mirrors.huaweicloud.com/python/3.10.9/python-3.10.9-amd64.exe
注意:
Windows下安装Python时,请务必勾选“Add Python to PATH”,以后我们调度Python时会方便很多。
如果你选择的是自定义安装,请务必在安装组件选择中勾选PIP(Python的包管理工具,无论是在接下来的安装中还是后续的使用中,我们都需要用到它)。
Nvidia CUDA(按需)
注意:如果CUDA版本不对,卸载对应版本后,删除C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA目录下的版本文件
https://developer.download.nvidia.com/compute/cuda/12.1.1/local_installers/cuda_12.1.1_531.14_windows.exe
下载并安装Microsoft C++生成工具(默认安装即可)(按需)
https://visualstudio.microsoft.com/zh-hans/visual-cpp-build-tools/
安装cuDNN cuDNN Archive | NVIDIA Developer
https://developer.download.nvidia.cn/compute/cudnn/secure/8.9.2/local_installers/12.x/cudnn-windows-x86_64-8.9.2.26_cuda12-archive.zip
安装 PyTorch 查看对应和安装命令