一、 基本环境
| 工具 | 版本号 |
|---|---|
| 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问题解决
