Cuda versions:

pip install torch torchvision \
    mxnet-cu102 \
    pandas \
    scikit-learn \
    jupyterlab \
    matplotlib \
    ipykernel \
    boto3