Thanks CostlyOstrich36 for you response. However, I am still getting errors after forcind add requirement via code Torch CUDA 116 download page found
clearml_agent: Warning: could not resolve python wheel replacement for torchvision!=0.13.0,>=0.8.1
clearml_agent: ERROR: Could not install task requirements!
Exception when trying to resolve python wheel: Could not find pytorch wheel URL for: torchvision!=0.13.0,>=0.8.1 with cuda 117 support