conda create -n ml-labs python=3.7
conda activate ml-labs

# for scikit-learn
pip install -U scikit-learn
# matplotlib
pip install matplotlib
# pandas
pip install pandas

# pytorch
# on cpu
pip install torch torchvision
# on gpu
# download following files from the link, please make sure the cuda version is supported by your driver
# https://download.pytorch.org/whl/cu117/torchvision-0.14.1%2Bcu117-cp37-cp37m-win_amd64.whl
# https://download.pytorch.org/whl/cu117/torch-1.13.1%2Bcu117-cp37-cp37m-win_amd64.whl
# change to your directory containing the 2 whl files, then run
pip install ./torch-1.13.1+cu117-cp37-cp37m-win_amd64.whl
pip install ./torchvision-0.14.1+cu117-cp37-cp37m-win_amd64.whl
# using conda to install pytorch is too much pain!
# you can test the installment of pytorch through pytorch_installation_test.py

pip install gpustat 
gpustat -cp -i

# watch -n 0.2 -d nvidia-smi is not compatible on windows