TextZoom icon indicating copy to clipboard operation
TextZoom copied to clipboard

Published 20 hours ago • verified publisher icon WenjiaWang0312

Training environment

Open AlexLi-98 opened this issue 4 years ago • 2 comments

Hi, could you please provide with a scirpt of needed packages and some information about training, like the version of PyTorch? I am also curious about the hardware(GPU) of your computer... Thanks in advance!

AlexLi-98 avatar Sep 29 '20 17:09 AlexLi-98

torch>=1.0.1 torchvision -c pytorch lmdb>=0.98 pillow>=7.1.2 numpy six tqdm python-opencv easydict yaml

I think most of them do not require specific versions.

WenjiaWang0312 avatar Nov 04 '20 09:11 WenjiaWang0312

which CUDA drivers should be installed? I am running CUDA 10.1, and have errors.

yustiks avatar Jan 24 '21 10:01 yustiks