Hi, welcome to PyTorch world.
Tensorflow’s installation guide 140’s NVIDIA requirements to run TensorFlow with GPU support would be helpful.  It tells you how to install CUDA etc.  Edited by moderator, pytorch does not need manual CUDA install
Then choose PyTorch with CUDA here 1.7k.