message: Add PyTorch dependencies
diff: @@ -16,6 +16,9 @@ install_requires = [
  "Pillow==6.1.0",
  "gym[atari,box2d]==0.13.1",
  "requests==2.22.0",
  + "torch==1.1.0.post2",
  + "torchvision==0.3.0",
  + "torchsummary==1.5.1",
  ]
  cpu_require = ["tensorflow==1.14.0"]
  