使用pytorch训练DCGAN----壹

Weach

已于 2024-05-04 21:41:52 修改

阅读量1.6k

点赞数

CC 4.0 BY-SA版权

分类专栏：笔记文章标签： pytorch 人工智能 python

于 2021-02-04 17:56:59 首次发布

本文链接：https://blog.csdn.net/qq_45748404/article/details/113654197

笔记专栏收录该内容

9 篇文章

订阅专栏

本文记录了一位开发者在使用PyTorch训练DCGAN时遇到的数据集准备、线程错误、Jupyter工作路径更改及虚拟环境配置等问题。通过详细步骤，最终成功解决所有问题，开始训练模型。尽管训练时间较长，但看到了初步的训练成果。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

欢迎大家来看我下一篇：使用pytorch训练DCGAN----贰(代码解析)

学习目标：

使用pytorch训练DCGAN（教程，源码均来自pytorch官方）

学习过程记录：

在训练过程中遇到的问题

1、数据集准备
Celeb-A Faces数据集【Download】数据集将下载为名为img_align_celeba.zip的文件。下载后，创建一个名为celeba的目录，并将zip文件解压缩到该目录中。这个目录不要弄错了！！！具体我的是

我的工作路径 E:\Pythonspace\pytorch\DCGAN.ipynb
我的数据集路径 E:\Pythonspace\pytorch\data\celeba

2、代码直接搬运到pycharm运行,提示线程错误

in the main module:

            if __name__ == '__main__':
                freeze_support()
                ...

        The "freeze_support()" line can be omitted if the program
        is not going to be frozen to produce an executable.

进程错误？按理说没问题啊！！！环境什么的我都有。
难道要用Jupyter？那改Jupyter吧！

3、 Jupyter工作路径更改
win10安装anaconda后Jupyter的默认工作路径应该是在C盘下，由于存放数据集等原因需更改（也可以说是创建一个新的工作路径，默认的并没有删除）~~C盘大的可忽略[dog]~~
具体的我参考的是这篇文章

4、 Jupyter中使用虚拟环境
兴高采烈打开Jupyter—》new notebook—》ctrl c + ctrl v + shift enter

---------------------------------------------------------------------------
ModuleNotFoundError                       Traceback (most recent call last)
<ipython-input-1-eb42ca6e4af3> in <module>
----> 1 import torch
ModuleNotFoundError: No module named 'torch'
---------------------------------------------------------------------------

？？？？没有我用anaconda创建的虚拟环境！！！！
~~众所周知~~ 要Jupyter Notebook支持虚拟运行环境，需要在Anaconda里安装一个插件参考。
打开Anaconda Promt，一般是（base）需要注意的是这里要激活你创建的虚拟环境conda activate '你的环境'然后执行conda install nb_conda，安装好后执行jupyter notebook打开Jupyter再new下你就可以看到有环境了在这里插入图片描述