zero-dce环境配置遇到的问题_ubuntu zero dce

1.服务器选择

笔者一开始使用自己上古老爷机配置环境，环境配置好了之后提示显存不够，笔者的老爷机用的是1050ti，2gb显存，迫于无奈，使用了autodl的云服务器，用了之后表示真香。

云服务器配置

镜像PyTorch  2.0.0 Python  3.8(ubuntu20.04) Cuda  11.8

GPU RTX 4090(24GB) * 1

CPU22 vCPU AMD EPYC 7T83 64-Core Processor

内存90GB

硬盘 系统盘:30 GB 数据盘:免费:50GB   付费:0GB

费用 ￥2.58/时（学生优惠）

2.conda环境配置

无论如何还是用一下conda，虽然是云服务器，但是还是喜欢用conda

conda create -n zero-dce python=3.6

笔者测试了各个python版本，3.6版本是笔者唯一配通的，其他版本应该也行，但是即不懒人，也不小白，不想试了，有能用的就挺好。

conda install pytorch torchvision torchaudio pytorch-cuda=11.8 -c pytorch -c nvidia

这个指令安装的几个核心安装包列出一下

 cudatoolkit-11.3.1         |       ha36c431_9       815.2 MB  nvidia
 numpy-1.19.2               |   py36h54aff64_0          22 KB  https://mirrors.ustc.edu.cn/anaconda/pkgs/main
 pytorch-1.10.2             |py3.6_cuda11.3_cudnn8.2.0_0        1.21 GB  pytorch
 pytorch-cuda-11.8          |       h7e8668a_5           3 KB  pytorch
 torchvision-0.11.3         |       py36_cu113        30.4 MB  pytorch

conda install opencv==3.4.2

这里一定要标明版本号，不然他会按默认的3.4.1，这个版本安装不上

可以进入python测试一下

>>> import torch
>>> print(torch.cuda.is_available())
True
>>> import cv2
>>>

3.测试代码

最后进入Zero-Dce_code文件夹，执行python lowlight_test.py命令

出现以下结果就是跑通了

data/test_data/DICM/62.jpg
0.0011279582977294922
data/test_data/DICM/60.jpg
0.0010986328125
data/test_data/DICM/48.jpg
0.0010895729064941406
data/test_data/DICM/49.jpg
0.0010881423950195312
data/test_data/DICM/61.jpg
0.0016222000122070312
data/test_data/DICM/65.jpg
0.0016405582427978516
data/test_data/DICM/64.jpg
0.0011129379272460938
data/test_data/DICM/58.jpg
0.0010781288146972656
data/test_data/DICM/66.jpg
0.0010840892791748047
data/test_data/DICM/67.jpg
0.0010845661163330078
data/test_data/DICM/14.jpg
0.0016644001007080078
data/test_data/DICM/28.jpg
0.0010845661163330078
data/test_data/DICM/29.jpg
0.003294706344604492
data/test_data/DICM/15.jpg
0.0010952949523925781
data/test_data/DICM/01.jpg
0.0016181468963623047
data/test_data/DICM/17.jpg
0.0010840892791748047
data/test_data/DICM/03.jpg
0.001104593276977539
data/test_data/DICM/02.jpg
0.0010933876037597656
data/test_data/DICM/16.jpg
0.0010840892791748047
data/test_data/DICM/12.jpg
0.0011096000671386719
data/test_data/DICM/06.jpg
0.0010859966278076172
data/test_data/DICM/07.jpg
0.0011060237884521484
data/test_data/DICM/13.jpg
0.001087188720703125
data/test_data/DICM/39.jpg
0.0016515254974365234
data/test_data/DICM/05.jpg
0.0010929107666015625
data/test_data/DICM/11.jpg
0.0010900497436523438
data/test_data/DICM/10.jpg
0.0010974407196044922
data/test_data/DICM/04.jpg
0.0010983943939208984
data/test_data/DICM/38.jpg
0.0016133785247802734
data/test_data/DICM/21.jpg
0.0011239051818847656
data/test_data/DICM/35.jpg
0.0016145706176757812
data/test_data/DICM/09.jpg
0.0010814666748046875
data/test_data/DICM/08.jpg
0.0011334419250488281
data/test_data/DICM/34.jpg
0.0029854774475097656
data/test_data/DICM/20.jpg
0.0011031627655029297
data/test_data/DICM/36.jpg
0.001613616943359375
data/test_data/DICM/22.jpg
0.0011951923370361328
data/test_data/DICM/37.jpg
0.0010869503021240234
data/test_data/DICM/33.jpg
0.0016956329345703125
data/test_data/DICM/27.jpg
0.0010914802551269531
data/test_data/DICM/26.jpg
0.0011241436004638672
data/test_data/DICM/32.jpg
0.0016918182373046875
data/test_data/DICM/18.jpg
0.0010998249053955078
data/test_data/DICM/30.jpg
0.0016639232635498047
data/test_data/DICM/31.jpg
0.0011620521545410156
data/test_data/DICM/25.jpg
0.0011000633239746094
data/test_data/DICM/19.jpg
0.0010790824890136719
data/test_data/DICM/42.jpg
0.0019576549530029297
data/test_data/DICM/56.jpg
0.0010945796966552734
data/test_data/DICM/57.jpg
0.0010800361633300781
data/test_data/DICM/43.jpg
0.001676797866821289
data/test_data/DICM/55.jpg
0.001127481460571289
data/test_data/DICM/41.jpg
0.001089334487915039
data/test_data/DICM/69.jpg
0.0010869503021240234
data/test_data/DICM/40.jpg
0.001680135726928711
data/test_data/DICM/54.jpg
0.001119375228881836
data/test_data/DICM/50.jpg
0.0010862350463867188
data/test_data/DICM/44.jpg
0.0016198158264160156
data/test_data/DICM/45.jpg
0.0010883808135986328
data/test_data/DICM/47.jpg
0.0010750293731689453
data/test_data/DICM/53.jpg
0.001089334487915039
data/test_data/DICM/52.jpg
0.001100301742553711
data/test_data/DICM/46.jpg
0.0010805130004882812
data/test_data/LIME/3.bmp
0.0017139911651611328
data/test_data/LIME/2.bmp
0.0015811920166015625
data/test_data/LIME/1.bmp
0.0016331672668457031
data/test_data/LIME/5.bmp
0.005883932113647461
data/test_data/LIME/4.bmp
0.0018324851989746094
data/test_data/LIME/6.bmp
0.0016851425170898438
data/test_data/LIME/7.bmp
0.0016520023345947266
data/test_data/LIME/9.bmp
0.001562356948852539
data/test_data/LIME/8.bmp
0.0016434192657470703
data/test_data/LIME/10.bmp
0.0017256736755371094

4.训练问题

在训练中出现了No module named ‘pytorch_colors‘报错，参考这篇博客可以解决

如何解决PyTorch中的No module named ‘pytorch_colors‘问题_colormath no module_花里梦雨的博客-CSDN博客