PyTorch test Before using the code, first invoke conda env source activate pytorch_env # where pytorch_env we created for pytorch