Windows10安装pytorch2.0.0(GPU)+pytorch-lightning2.0.0
安装stable diffusion时pytorch版本和torchvision不兼容,踩了很深的坑。从知乎找了到比较一体的解决方案,
记录如下:
引用:https://zhuanlan.zhihu.com/p/614624869
1. 新建conda的虚拟环境(将如下的xxx改为你自己的虚拟环境名)
conda create -n xxx python=3.10.6
2. 安装pytorch-lightning,使用了比较快的中科大源
python -m pip install -i https://pypi.mirrors.ustc.edu.cn/simple/ lightning
3. 查看Pytorch稳定版本,使用pip命令安装GPU版的Pytorch、匹配的Torchvison等(使用--trusted-host,禁用 SSL/TLS 证书验证,避免使用VPN等情况时的安装异常)
pip install torch==2.0.0+cu118 torchvision==0.15.1+cu118 -f https://download.pytorch.org/whl/torch_stable.html --trusted-host=pypi.python.org --trusted-host=pypi.org --trusted-host=files.pythonhosted.org