Jittor 热身赛

简介

本项目包含了第四届计图挑战赛热身赛的代码实现。

安装

本项目可在 1 张 LAPTOP GTX3060 上运行，训练时间约为 20 分钟。

运行环境

• Windows 11
• python >= 3.9
• jittor >= 1.3.8

安装依赖

执行以下命令安装 python 依赖

sudo apt install python3.7-dev libomp-dev
python -m pip install jittor
python -m jittor.test.test_example
python -m jittor.test.test_cudnn_op

训练

单卡训练可运行以下命令：

python CGAN.py