- Linux os (Ubuntu 18.04 or 20.04), Nvidia GPU (e.g. 2080ti) , Install GPU Driver
- Use Anaconda or Miniconda to build python environment. The Miniconda version I used can be downloaded here.
- Install CUDA in conda command window :
conda install cudatoolkit=10
, Then install cudnn :conda install cudnn=7.6.0
- Install other requirements by pip command, such as
pip install tensorflow-gpu==1.15.0
,pip install tqdm
,pip install glob
and so on. - Place the downloaded dataset in the corresponding directory, and then perform training and testing.