Thank. Gonna try that out. But i hit another snag. Strangely, the Agent is not creating the right venv. This is what the Agent created.
` pip:

• asn1crypto==0.24.0
• attrs==20.3.0
• certifi==2020.12.5
• chardet==4.0.0
• cryptography==2.1.4
• Cython==0.29.22
• furl==2.1.0
• future==0.18.2
• humanfriendly==9.1
• idna==2.6
• importlib-metadata==3.7.0
• jsonschema==3.2.0
• keyring==10.6.0
• keyrings.alt==3.0
• orderedmultidict==1.0.1
• pathlib2==2.3.5
• psutil==5.8.0
• pycrypto==2.6.1
• pygobject==3.26.1
• pyhocon==0.3.57
• PyJWT==1.7.1
• pyparsing==2.4.7
• pyrsistent==0.17.3
• python-dateutil==2.8.1
• pyxdg==0.25
• PyYAML==5.3.1
• requests==2.25.1
• requests-file==1.5.1
• SecretStorage==2.3.1
• six==1.11.0
• tqdm==4.54.1
• typing==3.7.4.3
• typing-extensions==3.7.4.3
• urllib3==1.26.3
• virtualenv==16.7.10
• zipp==3.4.0 But this is my requirements.txt attrs==20.3.0
  boto3==1.17.17
  botocore==1.20.17
  certifi==2020.12.5
  chardet==4.0.0
  clearml==0.17.4
  furl==2.1.0
  future==0.18.2
  humanfriendly==9.1
  idna==2.10
  jmespath==0.10.0
  jsonschema==3.2.0
  numpy
  orderedmultidict==1.0.1
  pathlib2==2.3.5
  Pillow==8.1.0
  psutil==5.8.0
  PyJWT==2.0.1
  pyparsing==2.4.7
  pyrsistent==0.17.3
  python-dateutil==2.8.1
  PyYAML==5.4.1
  requests==2.25.1
  requests-file==1.5.1
  s3transfer==0.3.4
  six==1.15.0
  torch==1.7.1
  torchvision==0.8.2
  typing-extensions==3.7.4.3
  urllib3==1.26.3 In particular, i am getting a error as follows. Environment setup completed successfully

Starting Task Execution:

Traceback (most recent call last):
File "pytorch_mnist.py", line 8, in <module>
import torch
ModuleNotFoundError: No module named 'torch'
DONE: Running task '3a90802d1dfa4ec09fbccba0beffbaa8', exit status 1 `