pytorch安装VAE项目详解

一、 基本环境

工具	版本号
OS	win 11
pycharm	2020.1
GPU	3050

二、代码来源

github地址为：

https://github.com/AntixK/PyTorch-VAE/blob/8700d245a9735640dda458db4cf40708caf2e77f/run.py

1.在pycharm的terminal输入：

git clone https://github.com/AntixK/PyTorch-VAE

下载项目
2.通过：

cd PyTorch-VAE

进入项目文件夹

三、搭建conda环境

1.点击Windows 安装 CUDA/cuDNN
2.配置conda 虚拟环境 Windows环境下anaconda创建python虚拟环境并配置到pycharm
配置环境使用的安装包及版本

工具	版本号
torch	1.13.1+cu117
pytorch-lightning	1.7.1
test-tube	0.6.6

切记torch 不要重新安装！！！！
其他的都按照requirements.txt文件来安装

pip install -r requirements.txt

四、下载数据集

这个项目无法自动下载数据集，需要手动安装
这个链接CelebA数据集详细介绍及其属性提取源代码有相关介绍，包括数据下载百度云地址
下载后修改文件的目录

提示：如果不修改代码会报错

五、启动项目

python run.py -c configs/vae.yaml

六、其他相关问题

1.pytorch报错torch.cuda.is_available()结果false处理方法
2.pytorch_lightning报错 You requested gpu: [1]，But your machine only has: [0]
3.问题：RuntimeError: Distributed package doesn‘t have NCCL built in
4.ForkingPickler(file, protocol).dump(obj)BrokenPipeError: [Errno 32] Broken pipe问题解决