conda install pytorch torchvision torchaudio cudatoolkit=11.3 -c pytorch
conda install -c anaconda pandas
conda install -c anaconda scikit-learn
conda install -c conda-forge tqdm
conda install -c anaconda numpy
conda install -c conda-forge matplotlib


