module load anaconda
conda create -n pytorch
source activate pytorch
conda install pytorch torchvision -c soumith
pip install gym
pip install gym[atari]
conda install matplotlib
conda install ipykernel
python -m ipykernel install --user --name pytorch --display-name "Python (pytorch)"
conda list
source deactivate