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

 

posted on 2023-05-14 17:06  漫步星野  阅读(686)  评论(0)    收藏  举报

导航